netron tensorboardx 可视化工具使用
1.下载安装netron和tensorboardx工具
pip install netron
pip install tensorboard
2.将搭建好的模型通过接口保存起来
d = torch.rand(1, 1, 28, 28)
m = AlexNet()#AlexNet网络结果模型
o = m(d)
onnx_path = "onnx_model_name.onnx"
torch.onnx.export(m, d, onnx_path)
3.通过netron命令访问该文件并在浏...
原创
2021-06-24 11:16:25 ·
187 阅读 ·
0 评论