model.eval

叶庭云3 个月前
pytorch·深度神经网络·dropout·model.eval·评估模式
分析为什么在 PyTorch 中，训练好深度神经网络后要使用 model.eval()🍉 CSDN 叶庭云：https://yetingyun.blog.csdn.net/训练模式 VS 评估模式。首先，我们需要明确 PyTorch 中的模型存在两种重要模式：训练模式（training mode）与评估模式（evaluation mode）。通过调用 model.eval() 方法，我们可以轻松地将模型切换到评估模式。