模型训练模式，预测模式-train() 和 eval()

转载：https://blog.csdn.net/weixin_43593330/article/details/107547202 

一、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()