我正在 tensorboardX 中检查我的模型。我正在使用 PyTorch。以下是forward
方法:
def forward(self, x):
x = self.net(x)
x = x.view(x.shape[0], -1) # flat
x = self.ext_net(x)
x = F.log_softmax(x, dim=1)
return x
在图中,我不明白蓝色和绿色框中的这些值和节点代表什么。input
为什么从to ext_net
( FCNCustom )有单独的数据流?不应该只有一个通过net
吗?