Module: tf.keras.models.experimental

Public API for tf.keras.models.experimental namespace.

Classes

class SharpnessAwareMinimization: Sharpness aware minimization (SAM) training flow.