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

on_batch_begin(
    batch,
    logs=None
)

on_batch_end

on_batch_end(
    batch,
    logs=None
)

on_epoch_begin

on_epoch_begin(
    epoch,
    logs=None
)

on_epoch_end

on_epoch_end(
    epoch,
    logs=None
)

on_train_batch_begin

on_train_batch_begin(
    batch,
    logs=None
)

on_train_batch_end

on_train_batch_end(
    batch,
    logs=None
)

on_train_begin

on_train_begin(logs=None)

on_train_end

on_train_end(logs=None)

set_model

set_model(model)

set_params

set_params(params)