如何在Pytorch中可视化网络？

Question

import torch
import torch.nn as nn
import torch.optim as optim
import torch.utils.data as data
import torchvision.models as models
import torchvision.datasets as dset
import torchvision.transforms as transforms
from torch.autograd import Variable
from torchvision.models.vgg import model_urls
from torchviz import make_dot

batch_size = 3
learning_rate =0.0002
epoch = 50

resnet = models.resnet50(pretrained=True)
print resnet
make_dot(resnet)

我想resnet从pytorch模型中形象化。我该怎么做？我尝试使用，torchviz但出现错误：

'ResNet' object has no attribute 'grad_fn'

Answer 1

make_dot需要一个变量（即带有的张量grad_fn），而不是模型本身。
尝试：

x = torch.zeros(1, 3, 224, 224, dtype=torch.float, requires_grad=False)
out = resnet(x)
make_dot(out)  # plot graph of variable, not of a nn.Module