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

html5 • • 问答

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

回答

来自Pytorch 论坛，稍作调整：

用

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

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

模式说明

以上是如何在Pytorch中检查模型是否处于训练或评估模式？的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>