TorchSummaryX 可視化卷積核，特徵圖尺寸步驟

我發現好像沒有人使用torchsummaryX 功能包進行卷積神經網絡的可視化 https://github.com/nmhkahn/torchsummaryX 強烈推薦，最近在看Attention Unet 網絡架構，始終不知道卷積核大小和特徵圖大小。但是安裝該工具包之後，還是有點一目瞭然。 1.安裝pytorch. https://pytorch.org/ 在pytorch 官網中找自己的下載

>>阅读原文<<