我正在使用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'])