module的内部包含一个training的属性（bool类型），调用training即可。
if net.training
xxx
else:
yyy
多说一句：一般我们调用net.eval() 或者net.train()都会对这个training属性进行改写。
Pytorch--判断model是否在训练模式下
这篇博客探讨了在深度学习中，module内部的training属性如何影响模型的行为。通过检查net.training的状态，可以决定执行xxx操作（在训练模式下）还是yyy操作（在评估模式下）。通常，调用net.eval()或net.train()会动态改变这个属性，以适应不同的运行阶段。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
