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

Question

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

Answer 1

来自Pytorch论坛，有一个小调整：

使用

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

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

模式说明