機器學習入門

一、學習

臺灣大學李宏毅教授的《機器學習》課程

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，就是你的圖片的參數