用pytorch搭建AlexNet（微调预训练模型及手动搭建）

本文介绍了如何在pytorch下搭建AlexNet，使用了两种方法，一种是直接加载预训练模型，并根据自己的需要微调（将最后一层全连接层输出由1000改为10），另一种是手动搭建。 构建模型类的时候需要继承自torch.nn.Module类，要自己重写 _ _ \_\_ __init _ _ \_\_ __方法和正向传递时的forward方法，这里我自己的理解是，搭建网络写在 _ _ \_\_ __

>>阅读原文<<