netron 是一个非常好用的网络结构可视化工具。
使用方法:
pip install netron |
但是netron对pytorch模型的支持还不成熟。自己试的效果是生成的模型图没有连线。
这里就有一个把 .pth
模型转化为 .onnx
模型。
Pytorch模型转onnx
import torch |
可视化onnx模型
import netron |
效果图:
netron 是一个非常好用的网络结构可视化工具。
使用方法:
pip install netron |
但是netron对pytorch模型的支持还不成熟。自己试的效果是生成的模型图没有连线。
这里就有一个把 .pth
模型转化为 .onnx
模型。
import torch |
import netron |
效果图: