Pytorch中的train和eval模式详解

(一)、model.train()和model.eval()分别在训练和测试中都要写，它们的作用如下：

(1). model.train()
启用 BatchNormalization 和 Dropout，将BatchNormalization和Dropout置为True
(2). model.eval()
不启用 BatchNormalization 和 Dropout，将BatchNormalization和Dropout置为False

(二)、总结

(1). 在训练模块中千万不要忘了写model.train()
(2). 在评估（或测试）模块千万不要忘了写model.eval()
 

转自：https://blog.csdn.net/sinat_36618660/article/details/100147506