Pytorch可视化——Netron

原文地址

Pytorch系列目录

• Netron

  Netron官方/下载网址

  Netron GitHub

  Windows下安装Netron这个软件很简单，只需要进入上面的下载地址，点击“Download for Windows”即可下载安装包，双击安装包即可安装

  

  图片来源

  Netron的简单使用是通过打开保存好的模型文件，从上图可以看到，Netron目前可以支持的平台很多，在其中就有Pytorch，可以不必纠结于模型文件的后缀，我实践过这个软件是可以打开pytorch的.pkl的模型的

  我们知道pytorch是能保存整个模型和只保存参数的，这里只有打开完整的模型才获得一个比较好的显示效果，打开参数则只能显示部分

  • 打开一个完整模型

    

    点击层名称，可以弹出右边的详细信息

  • 只读取模型参数文件

    

    这个图跟上面的图来自同一个模型，只是前者是保存的整个模型，后者只保存了模型的参数

  • 图中模型程序

    程序地址