TensorFlow回調(diào)函數(shù)：tf.keras.callbacks.LearningRateScheduler

tf.keras.callbacks.LearningRateScheduler函數(shù)

類(lèi)LearningRateScheduler

繼承自：Callback

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

學(xué)習(xí)速率調(diào)度程序。

參數(shù)：

• schedule：一個(gè)函數(shù)，它將一個(gè)epoch索引作為輸入（整數(shù)，從0開(kāi)始索引）并返回一個(gè)新的學(xué)習(xí)速率作為輸出（浮點(diǎn)數(shù)）。
• verbose：int；當(dāng)其為0時(shí)，保持安靜；當(dāng)其為1時(shí)，表示更新消息。

__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)