用pytorch搭建AlexNet（微調預訓練模型及手動搭建）

本文介紹瞭如何在pytorch下搭建AlexNet，使用了兩種方法，一種是直接加載預訓練模型，並根據自己的需要微調（將最後一層全連接層輸出由1000改爲10），另一種是手動搭建。 構建模型類的時候需要繼承自torch.nn.Module類，要自己重寫 _ _ \_\_ __init _ _ \_\_ __方法和正向傳遞時的forward方法，這裏我自己的理解是，搭建網絡寫在 _ _ \_\_ __

>>阅读原文<<