我正在使用Google Colab来学习一些卷积神经网络(CNN)。
我使用model.compile()来设置我的损失函数和优化器函数。
在以下代码中,我应该在哪里修改学习率呢?
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
回答:
你可以不传递字符串,而是传递一个优化器到compile方法中,并如以下所示设置优化器的学习率:
from keras import optimizersoptm = 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'])