Pytorch可视化——Netron

  • Netron

    Netron官方/下载网址

    Netron GitHub

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

    NETRON

    图片来源

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

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

    • 打开一个完整模型

    完整模型

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

    • 只读取模型参数文件

    参数

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

    • 图中模型程序

    程序地址

You may also like...

发表评论

电子邮件地址不会被公开。 必填项已用*标注