model.train() && model.eval()

1、model.train

model.train()是在模型训练的时候使用,主要针对Batch Normalization 和 Dropout 方法模式,防止网络过拟合。

2、model.eval

model.eval()是在模型检验的时候使用的,也是针对BN与Dropout方法模式的,pytorch会自动把BN和DropOut固定住,不会取平均,而是用训练好的值。具体目的大概是为了防止图片失真。

因此,在使用Pytorch训练和评价模式的时候,一定要把实例化的model指定为对应的模式。

猜你喜欢

转载自blog.csdn.net/xdg15294969271/article/details/121738675
今日推荐