在卷积神经网络中设置学习率

Question

我正在使用Google合作来学习一些CNN。

我正在使用model.compile（）设置损失和优化器功能。

在以下代码中，我在哪里更改学习率？

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

Answer 1

代替传递字符串，您可以传递Room来编译方法并将学习率设置为优化器，如下所示：

from keras import optimizers
optm = optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=False)
model.compile(optimizer=optm,
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])