1.打印出 络的参数量 , 直接打出 络结构(不够直观)
.
2. 利用可视化工具,直观显示 络结构
Netron是一款可视化 络模型的软件,目前的Netron支持主流各种框架的模型结构可视化工 https://github.com/lutzroeder/Netron
支持的文件类型如下:
点开链接后,按自己的系统直接点击下载就可以了,比如我用的win10,就按如下下载:
安装后,直接把 络文件(如onnx)直接拖入就能显示了:
比如,我常用的就是pytorch,那么首先把pytorch转为onnx文件:
再拖进去就显示啦!
声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!