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

Question

How to check from within a model if it is currently in train or eval mode?

Answer 1

From the Pytorch forum , with a small tweak:

use

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

Always better to have a stack overflow answer than to look at forums.

Explanation about the modes