可視化網絡模型結構

第一種：查看參數量 pip install torchsummary #此段代碼之顯示Output Shape和Param from torchsummary import summary summary(model.cuda(), input_size=(3, 224, 224)) 224分別爲輸入高度和寬度，3爲通道數，界面如下所示 第2種：查看參數量，這種的最詳細，但是也是最新的方法，所以還

>>阅读原文<<