netron是微软小哥lutzroeder的一个广受好评的开源项目地址:
可惜,默认支持的格式中并不包括pytorch可能当年小哥面试facebook被拒了,:)
由于不支持默认的pytorch模型格式(.pth)因此需要存为onnx,庆幸pytorch支持!
執行上面代码后会调用本地浏览器打开,形式和tensorboard差不多
netron是微软小哥lutzroeder的一个广受好评的开源项目地址:
可惜,默认支持的格式中并不包括pytorch可能当年小哥面试facebook被拒了,:)
由于不支持默认的pytorch模型格式(.pth)因此需要存为onnx,庆幸pytorch支持!
執行上面代码后会调用本地浏览器打开,形式和tensorboard差不多
版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。