tf.keras.optimizers.schedules.PiecewiseConstantDecay

TensorFlow 2 version View source on GitHub

A LearningRateSchedule that uses a piecewise constant decay schedule.

Inherits From: LearningRateSchedule

tf.keras.optimizers.schedules.PiecewiseConstantDecay(
    boundaries, values, name=None
)

boundaries A list of Tensors or ints or floats with strictly increasing entries, and with all elements having the same type as the optimizer step.
values A list of Tensors or floats or ints that specifies the values for the intervals defined by boundaries. It should have one more element than boundaries, and all elements should have the same type.
name A string. Optional name of the operation. Defaults to 'PiecewiseConstant'.

ValueError if the number of elements in the lists do not match.

Methods

from_config

View source

@classmethod
from_config(
    config
)

Instantiates a LearningRateSchedule from its config.

Args
config Output of get_config().

Returns
A LearningRateSchedule instance.

get_config

View source

get_config()

__call__

View source

__call__(
    step
)

Call self as a function.