pip install netron
import netron
modelpath = "model.pt" # 定义模型数据保存的路径
# onnx,pth都可以
netron.start(modelpath) # 输出网络结构
pip install netron
import netron
modelpath = "model.pt" # 定义模型数据保存的路径
# onnx,pth都可以
netron.start(modelpath) # 输出网络结构