只打印训练阶段的损失信息

在 PyTorch 训练模型时,你可能只想打印每个 epoch 的损失,而不是其他信息,例如训练准确率。以下是如何修改代码以只打印损失:

print('Epoch: %d, Loss: %.3f' % (epoch + 1, running_loss))

这个修改保留了 'Epoch' 和 'Loss' 部分,并使用字符串格式化将 epoch 编号和损失值打印出来。

修改后的代码将只打印每个 epoch 的损失,例如:

Epoch: 1, Loss: 2.301
Epoch: 2, Loss: 1.875
Epoch: 3, Loss: 1.562
...

希望这个修改能满足你的需求!如果你还有其他问题,请随时提问。

PyTorch训练过程:只显示损失信息

原文地址: https://www.cveoy.top/t/topic/NnO 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录