前提安装netron、torch、torchvision包
执行下方代码即可
# 针对有网络模型,但还没有训练保存 .pth 文件的情况
import netron
import torch.onnx
from torch.autograd import Variable
from torchvision.models import resnet50 # 以 resnet18 为例
myNet = resnet50() # 实例化 resnet18
x = torch.randn(16, 3, 40, 40) # 随机生成一个输入
modelData <