pytorch中net.eval() 和net.train()的使用

参考链接：https://www.jianshu.com/p/822d9ae0169d

---

神经网络模块存在两种模式，train模式（net.train())和eval模式（net.eval())。一般的神经网络中，这两种模式是一样的，只有当模型中存在dropout和batchnorm的时候才有区别。

一旦我们用测试集进行结果测试的时候，一定要使用net.eval()把dropout关掉，因为这里我们的目的是测试训练好的网络，而不是在训练网络，没有必要再dropout和再计算BN的方差和均值(BN使用训练的历史值)。

原文链接：https://blog.csdn.net/vectorindex/article/details/115554228