如何在 Pytorch 中检查模型是处于训练模式还是评估模式？
Posted
技术标签:
【中文标题】如何在 Pytorch 中检查模型是处于训练模式还是评估模式？【英文标题】：How to check if a model is in train or eval mode in Pytorch? 【发布时间】：2021-03-28 08:55:56 【问题描述】：如何从模型中检查它当前是否处于训练或评估模式？
【问题讨论】：
【参考方案1】：来自the Pytorch forum，稍作调整：
使用
if self.training:
# it's in train mode
else:
# it's in eval mode
获得堆栈溢出答案总是比查看论坛更好。
Explanation about the modes
【讨论】：
以上是关于如何在 Pytorch 中检查模型是处于训练模式还是评估模式？的主要内容，如果未能解决你的问题，请参考以下文章
Pytorch中如何使用DataLoader对数据集进行批训练