tf.keras.experimental.CosineDecay