当前位置:   article > 正文

Pytorch可视化——Netron_netron download

netron download

原文地址

Pytorch系列目录

  • Netron

    Netron官方/下载网址

    Netron GitHub

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

    NETRON

    图片来源

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

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

    • 打开一个完整模型

      完整模型

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

    • 只读取模型参数文件

      参数

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

    • 图中模型程序

      程序地址

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/656102
推荐阅读
相关标签
  

闽ICP备14008679号