一、学习
台湾大学李宏毅教授的《机器学习》课程
https://www.bilibili.com/video/av10590361
二、调试
-
打印网络结构
先安装torchsummary工具,pip.exe install torchsummaryfrom torchsummary import summary model = VGG('VGG11').to(device) summary(model, input_size=(3, 32, 32)) #input_size中参数分别表示C,H,W,就是你的图片的参数