深度学习模型优化:学习率衰减与Adam优化器
配置模型
创建一个ReduceLROnPlateau回调函数,用于根据验证集上的损失减少学习率
reduce_lr = ReduceLROnPlateau(monitor='val_loss', factor=np.sqrt(0.1), patience=5)
创建一个Adam优化器,学习率为0.001
optimizer = Adam(learning_rate=0.001)
cnn_model.compile(optimizer=optimizer, loss=CategoricalCrossentropy(), metrics=['accuracy'])
原文地址: http://www.cveoy.top/t/topic/pggz 著作权归作者所有。请勿转载和采集!