TensorFlow回調函數：tf.keras.callbacks.LearningRateScheduler

tf.keras.callbacks.LearningRateScheduler函數

類LearningRateScheduler

繼承自：Callback

定義在：tensorflow/python/keras/callbacks.py。

學習速率調度程序。

參數：

• schedule：一個函數，它將一個epoch索引作為輸入（整數，從0開始索引）并返回一個新的學習速率作為輸出（浮點數）。
• verbose：int；當其為0時，保持安靜；當其為1時，表示更新消息。

__init__

__init__(
    schedule,
    verbose=0
)

初始化自我。

方法

on_batch_begin(
    batch,
    logs=None
)

on_batch_end(
    batch,
    logs=None
)

on_epoch_begin(
    epoch,
    logs=None
)

on_epoch_end(
    epoch,
    logs=None
)

on_train_batch_begin(
    batch,
    logs=None
)

on_train_batch_end(
    batch,
    logs=None
)

on_train_begin(logs=None)

on_train_end(logs=None)

set_model(model)

set_params(params)