Pytorch中的model.train()和model.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()