如何在Pytorch中检查模型是否处于训练或评估模式? [英] How to check if a model is in train or eval mode in Pytorch?

问题描述

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

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

推荐答案

来自使用

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.

关于模式的说明

这篇关于如何在Pytorch中检查模型是否处于训练或评估模式?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！