Pytorch可視化——Netron

原文地址

Pytorch系列目錄

• Netron

  Netron官方/下載網址

  Netron GitHub

  Windows下安裝Netron這個軟件很簡單，只需要進入上面的下載地址，點擊“Download for Windows”即可下載安裝包，雙擊安裝包即可安裝

  圖片來源

  Netron的簡單使用是通過打開保存好的模型文件，從上圖可以看到，Netron目前可以支持的平臺很多，在其中就有Pytorch，可以不必糾結於模型文件的後綴，我實踐過這個軟件是可以打開pytorch的.pkl的模型的

  我們知道pytorch是能保存整個模型和只保存參數的，這裏只有打開完整的模型才獲得一個比較好的顯示效果，打開參數則只能顯示部分

  • 打開一個完整模型

    點擊層名稱，可以彈出右邊的詳細信息

  • 只讀取模型參數文件

    這個圖跟上面的圖來自同一個模型，只是前者是保存的整個模型，後者只保存了模型的參數

  • 圖中模型程序

    程序地址