今天小编就为大家分享一篇使用pytorch搭建AlexNet操作(微调预训练模型及手动搭建)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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