关于Pytorch中model.train（）

model.train（）将模型设置为训练状态，作用：使Dropout，batchnorm知道后有不同表现（具体参考Dropout，batchnorm源码），只有这两个关心True or False。
将模型设置为测试状态有两种方法：
1.model.train(mode=False)
2.model.eval() 相当于第一种方法

model.train（）源码：

model.eval() 源码：

原文：https://www.cnblogs.com/wmc258/p/14520098.html