文章目录
1、model.train2、model.eval1、model.train
model.train()是在模型训练的时候使用,主要针对Batch Normalization 和 Dropout 方法模式,防止网络过拟合。
2、model.eval
model.eval()是在模型检验的时候使用的,也是针对BN与Dropout方法模式的,pytorch会自动把BN和DropOut固定住,不会取平均,而是用训练好的值。具体目的大概是为了防止图片失真。
因此,在使用Pytorch训练和评价模式的时候,一定要把实例化的model指定为对应的模式。