netron是微软小哥lutzroeder的一个广受好评的开源项目地址：

可惜，默认支持的格式中并不包括pytorch可能当年小哥面试facebook被拒了，:)

由于不支持默认的pytorch模型格式(.pth)因此需要存为onnx，庆幸pytorch支持！

執行上面代码后会调用本地浏览器打开，形式和tensorboard差不多