tf.keras.optimizers.schedules.serialize

Serializes a LearningRateSchedule into a JSON-compatible dict.

tf.keras.optimizers.schedules.serialize(
    learning_rate_schedule, use_legacy_format=False
)

learning_rate_schedule The LearningRateSchedule object to serialize.

A JSON-serializable dict representing the object's config.

Example:

lr_schedule = tf.keras.optimizers.schedules.ExponentialDecay(
  0.1, decay_steps=100000, decay_rate=0.96, staircase=True)
tf.keras.optimizers.schedules.serialize(lr_schedule)
{'module': 'keras.optimizers.schedules',
'class_name': 'ExponentialDecay', 'config': {...},
'registered_name': None}