ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf.keras.optimizers.schedules

Public API for tf.keras.optimizers.schedules namespace.

Classes

class CosineDecay: A LearningRateSchedule that uses a cosine decay schedule.

class CosineDecayRestarts: A LearningRateSchedule that uses a cosine decay schedule with restarts.

class ExponentialDecay: A LearningRateSchedule that uses an exponential decay schedule.

class InverseTimeDecay: A LearningRateSchedule that uses an inverse time decay schedule.

class LearningRateSchedule: The learning rate schedule base class.

class PiecewiseConstantDecay: A LearningRateSchedule that uses a piecewise constant decay schedule.

class PolynomialDecay: A LearningRateSchedule that uses a polynomial decay schedule.

Functions

deserialize(...): Instantiates a LearningRateSchedule object from a serialized form.

serialize(...): Serializes a LearningRateSchedule into a JSON-compatible representation.