机器学习入门

一、学习

台湾大学李宏毅教授的《机器学习》课程

https://www.bilibili.com/video/av10590361

二、调试

• 打印网络结构

  先安装torchsummary工具，pip.exe install torchsummary

  from torchsummary import summary
  model = VGG('VGG11').to(device)
  summary(model, input_size=(3, 32, 32)) #input_size中参数分别表示C,H,W，就是你的图片的参数