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

Question

19

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

- Gulzar

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Gulzar · Accepted Answer

从Pytorch论坛中，稍作调整：

使用

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

与其查看论坛，有一个堆栈溢出的答案总是更好的选择。