Pytorch实现AlexNet学习笔记

代码

model.train():

model.train()，训练过程中，作用：启用batch normalization和drop out。

model.eval()，测试过程中，作用：沿用batch normalization的值，不使用drop out