tf.keras.callbacks.LearningRateScheduler

Class LearningRateScheduler

Inherits From: Callback

Defined in tensorflow/python/keras/callbacks.py.

Learning rate scheduler.

Arguments:

  • schedule: a function that takes an epoch index as input (integer, indexed from 0) and returns a new learning rate as output (float).
  • verbose: int. 0: quiet, 1: update messages.

Methods

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

on_train_begin(logs=None)

on_train_end

on_train_end(logs=None)

set_model

set_model(model)

set_params

set_params(params)