pytorch：可视化模型结构

1、首先要能print出你的模型结构print(model)

2、将你的model模型结构保存成.onnx的文件

3、使用pytorch中的netron方法

# 针对有网络模型，但还没有训练保存 .onxx 文件的情况
import netron
import torch.onnx
from torch.autograd import Variable
from torchvision.models import resnet18  # 以 resnet18 为例

myNet = resnet18()  # 实例化 resnet18
x = torch.randn(16, 3, 40, 40)  # 随机生成一个输入
modelData = "./demo.onnx"  #  onnx 文件
torch.onnx.export(myNet, x, modelData)  # 将 pytorch 模型以 onnx 格式导出并保存
netron.start(modelData)  # 输出网络结构

结果如下图所示：