我的验证损失为1.86，我应该如何减少它？

Posted 2021-03-19

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

答案

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

我建议你仔细看一下前几个纪元的图，并提前停止训练，以保存验证损失最小的模型。数据增强可能也有助于延迟过拟合的发生。