["python","pytorch"]
How do I visualize a net in Pytorch?

Consider: 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) I want to visualize resnet from the PyTorch models. How can I do it? I tried to use torchviz, but it gives an error: 'ResNet' object has no attribute 'grad_fn'
The make_dot expects a variable (i.e., tensor with grad_fn), not the model itself. try: 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