我使用 LearningRateScheduler 进行模型训练。我想在 CSV 文件（或其他文档文件）中保存每个时期的学习率。
有没有办法使用回调来保存这些学习率？
如何使用回调在每个完成的时期保存我的学习率？
数据挖掘
计算机视觉
2022-03-13 18:38:30
1个回答
您可以编写自定义回调并将 LR 保存在文件中。
你会得到它 -self.model.optimizer.learning_rate
自定义回调 - Keras docs
class CustomCallback(keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs=None):
print("LR - {}".format(self.model.optimizer.learning_rate))
my_callbacks = [ CustomCallback() ]
LR - <tf.Variable 'Adam/learning_rate:0' shape=() dtype=float32, numpy=0.001 >
其它你可能感兴趣的问题