tensorflow - 我得到了 1.86 的验证损失，我应该如何减少它？

Question

我的模型图 我运行了一个模型来对两种不同类型的图像进行分类，我得到 1.875 的 val_loss，600 张图像的学习率为 1e-3 我该怎么办？

标签： tensorflow

Answer 1

从您提供的图表来看，您的模型似乎从 20 纪元开始过度拟合。这是有道理的，因为您只有 600 张图像。

我建议您仔细查看第一个 epoch 的图表，并尽早停止训练以保存具有最低验证损失的模型。数据增强也可能有助于延迟过拟合的发生。