1

我正在 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吗?

在此处输入图像描述

4

0 回答 0