pytochはネットワーク構造を調べます。

参照してください:http://www.freesion.com/article/340667237/
端末で自分達のネットワーク構造をpdfファイルに保存したいです。次のような方法を使います。

import torch
from torchvision.models import AlexNet
from torchviz import make_dot

x=torch.rand(8,1,224,224).cuda()
model=torch.load('/home/resnet_mnist.pth')
y=model(x)
g = make_dot(y)
g.render('./weights/espnet_model', view=False) 

ここの入力データはこのような形でなければいけません。x=torch.rand(8,1,224,224).cuda() # GPU .cuda()