设置卷积神经网络的学习率

Question

我正在使用 google collab 学习一些 CNN。

我正在使用 model.compile() 来设置我的损失函数和优化器函数。

在下面的代码中我在哪里改变学习率？

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

Answer 1

您可以将 optimizer 传递给编译方法并将您的学习率设置为优化器，而不是传递字符串，如下所示：

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'])

设置卷积神经网络的学习率

Setting the learning rate in convolutional neural network

machine-learning

keras

python-3.7