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

Question

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

Answer 1

来自Pytorch 论坛，稍作调整：

用

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

拥有堆栈溢出的答案总是比查看论坛更好。

模式说明