2.分享一个好看的pytorch可视化模型工具netron
作者:互联网
使用netron工具可视化Pytorch模型
1 安装netron
pip install netron
2 导入包
import netron
import torch.onnx
程序调用
if __name__ == '__main__':
net = vgg()
x = Variable(torch.FloatTensor(16, 3, 40, 40))
y = net(x)
print(y.data.shape)
onnx_path = "onnx_model_name.onnx"
torch.onnx.export(net, x, onnx_path)
netron.start(onnx_path)
展示图
标签:__,netron,onnx,torch,pytorch,可视化,path,net 来源: https://blog.csdn.net/qq_30093417/article/details/118338845