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

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

在以下代码中,我应该在哪里更改学习率?

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

最佳答案

除了传递字符串,您还可以传递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'])

08-25 00:12