在使用预训练模型来训练我的数据集时,如何调整学习率或其他参数以获得更好的效果?请给我一些建议
作者: capel888创建于 2023年3月22日更新于 2023年5月2日
我正在使用预训练的权重 BDD100K 来训练自己的数据集。您能给我一些建议吗？(所有参数均未变化，例如学习率和学习率衰减步长)以下是我 12 个 epoch 的训练结果。我认为地图太低了，但图中显示的每个损失函数的值都很小。我担心如果继续不做任何超参数调整，模型将会过拟合。目前我将训练轮数设置为 250 epoch，每 10 个轮次评估一次，每 50 个 epoch 保存一次模型。下图显示了我训练了 10 个 epoch 的损失函数以及在验证集上的评估结果
内容来源: eriklindernoren/PyTorch-YOLOv3