我应该为Keras中的任何优化程序的衰减参数设置什么值

时间:2019-08-08 06:50:45

标签: machine-learning keras deep-learning

我正在使用Keras中的转移学习训练图像分类模型。我正在使用Adam优化器,初始学习率为0.001。我想了解如何为decay参数设置什么值。我正在训练100个时代,并且使用的批量大小为16。

1 个答案:

答案 0 :(得分:2)

您只需要实例化Adam并在构造函数(__init__)中设置参数即可:

opt = Adam(lr=0.01, decay=0.99)
model.compile(loss=..., optimizer=opt)

参数的实际值是您必须尝试的东西,它是可以使用Grid或Random Search进行调整的超参数。一些常见的值为0.99和0.95。