TensorFlow回调函数:tf.keras.callbacks.LearningRateScheduler

2019-03-27 15:18 更新

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)
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号