如何在 Pytorch 中检查模型是处于训练模式还是评估模式？

【问题标题】：How to check if a model is in train or eval mode in Pytorch?如何在 Pytorch 中检查模型是处于训练模式还是评估模式？
【发布时间】：2021-03-28 08:55:56
【问题描述】：

如何从模型中检查它当前是否处于训练或评估模式？

【问题讨论】：

【解决方案1】：

来自the Pytorch forum，稍作调整：

使用

if self.training:
    # it's in train mode
else:
    # it's in eval mode

获得堆栈溢出答案总是比查看论坛更好。

【讨论】：