在使用Hidden Layer可视化的时候报这个错
代码:
import torch
import torchvision
torchvision 主要的三个包
models:提供了很多常用的训练好的网络模型,我们可以直接加载并使用,如Alexnet、ResNet等。
datasets:提供了(1)一些常用的图片数据集,如MNIST、COCO等(2)加载自己的数据集的常用方法,
目前只有DatasetFolder、ImageFolder、VisionDataset三个方法。
transforms:提供了一些常用的图像转换处理操作,主要针对Tensor或PIL Image进行操作
import hiddenlayer as h
定义网络
model = torchvision.models.alexnet()
定义输出
x = torch.randn([3,3,224,224])
可视化网络并将结果保存为图片
NetCraph = h.build_graph(model,x)
NetCraph.save('./model.png',format='png')
打印网络结构
print(model)
报错:
TypeError: 'torch._C.Node' object is not subscriptable
有没有人知道怎么解决