AttributeError: module 'keras.api._v2.keras.optimizers' has no attribute 'experimental'

Issue Type

Bug

Have you reproduced the bug with TF nightly?

Yes

Source

source

Tensorflow Version

'2.6.0'

Custom Code

Yes

OS Platform and Distribution

windows10

Mobile device

No response

Python version

3.9.12

Bazel version

No response

GCC/Compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current Behaviour?

A bug happened!
AttributeError: module 'keras.api._v2.keras.optimizers' has no attribute 'experimental'

Standalone code to reproduce the issue

AttributeError                            Traceback (most recent call last)
Cell In [34], line 1
----> 1 trained_model = run_experiment()

Cell In [33], line 8, in run_experiment()
      3 filepath = "C:/Users/HP/video_classifier"
      4 checkpoint = keras.callbacks.ModelCheckpoint(
      5     filepath, save_weights_only=True, save_best_only=True, verbose=1
      6 )
----> 8 model = get_compiled_model()
      9 history = model.fit(
     10     X_train,
     11     y_train,
   (...)
     14     callbacks=[checkpoint],
     15 )
     17 model.load_weights(filepath)

Cell In [32], line 21, in get_compiled_model()
     16     model = keras.Model(inputs, outputs)
     17     lr_schedule = keras.optimizers.schedules.ExponentialDecay(
     18     initial_learning_rate=1e-3,
     19     decay_steps=10000,
     20     decay_rate=0.9)
---> 21     opt = tf.keras.optimizers.experimental.AdamW(
     22     learning_rate=lr_schedule,
     23     beta_1=0.9,
     24     beta_2=0.999,
     25     epsilon=1e-08,
     26     weight_decay = 1e-04
     27 )
     28     model.compile(
     29         optimizer= opt, loss="sparse_categorical_crossentropy", metrics=["accuracy"]
     30     )
     31     return model

AttributeError: module 'keras.api._v2.keras.optimizers' has no attribute 'experimental'

Relevant log output

No response