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

Question

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

Answer 1

来自Pytorch 论坛，稍作调整：

利用

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

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

模式说明