第四篇 AlexNet——网络实战

在前面的文章我介绍了AlexNet的网络结构和数据集的制作，这篇文章我将和大家一起完成AlexNet模型的实战。数据集选用上篇制作的数据集，模型用Pytorh自带的AlexNet。在这篇文章中，我尽量的简化代码，只保留最基本的逻辑，让每一个初学者能够看明白。通过这篇文章你能学到：1、如何训练模型？2、如何推理？3、如何读取数据集、处理数据集？4、如何使用余弦退火调整学习率？5、如何保存权重文件和整个模型文件？