刚才第一次接触netron来可视化模型,然后出现的问题就是展示的图片中多了很多“identity”,我的模型是双分支的,代码如下:
def forward(self, x1, x2):
x1 = self.stem(x1)
x2 = self.stem(x2)
x1 = self.conv_block(x1)
x2 = self.conv_block(x2)
x = torch.cat((x1, x2), dim=-1)
# 后面的代码省略
可以看出上述代码两个输入x1,x2分别经过的是相同的处理,但是netron显示的图片如下,发现第二个分支多了很多identity,和第一个分支有明显的区别,请问这个是什么原因呢?能不能展示成两个分支一样的呢?