问题

最近学习pytorch，原来用kreas重现的模型改为用pytorch实现训练，因为这样给模型的操作更加细致，对模型的掌控更好。

当我写好一个模型出现了这个问题

使用pytorchviz进行模型可视化出现r如下错误

raceback (most recent call last):

  File "/home/jiwu/Documents/AttRCNN-CNNs/pyt_train.py", line 174, in <module>

    g = make_dot(y)

  File "/home/jiwu/.virtualenvs/jiwu/lib/python3.6/site-packages/torchviz/dot.py", line 37, in make_dot

    output_nodes = (var.grad_fn,) if not isinstance(var, tuple) else tuple(v.grad_fn for v in var)

AttributeError: 'NoneType' object has no attribute 'grad_fn'

找了很久没发现原因在哪里，翻pytorchviz的issue，一直google没找到原因，最后才发现是只是在foward 函数我没有return x。 所以出现了这个问题。

    def forward(self, x):

        print (x.shape)

        x1 = self.conv1(x)

        x2 = self.conv2(x)

        print (x2.shape)

        x3 = self.conv3(x)

        x4 = self.conv4(x)

        print (x1.shape, x2.shape, x3.shape, x4.shape)

        x = torch.cat((x1, x2, x3, x4), dim = 1)

        print (x.shape)

        x = x.view(x.size(0), -1)

        print(x.shape)

        x = self.line1(x)

        x = self.line2(x)

        x = self.line3(x)

        return x

巴特西

使用pytorchviz进行模型可视化出现 'NoneType' object has no attribute 'grad_fn'

问题

使用pytorchviz进行模型可视化出现r如下错误

最新文章

热门文章