Remove experimental optimizer usage on github.

PiperOrigin-RevId: 435102510

Remove experimental optimizer usage on github.
PiperOrigin-RevId: 435102510
af788dab · Hongkun Yu · A. Unique TensorFlower · 0f9f7c8e · af788dab
Commit af788dab authored Mar 16, 2022 by Hongkun Yu Committed by A. Unique TensorFlower Mar 16, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/modeling/optimization/optimizer_factory.py official/modeling/optimization/optimizer_factory.py +2 -2

No files found.
--- a/official/modeling/optimization/optimizer_factory.py
+++ b/official/modeling/optimization/optimizer_factory.py
@@ -28,9 +28,9 @@ from official.nlp import optimization as nlp_optimization

 OPTIMIZERS_CLS = {
    'sgd': tf.keras.optimizers.SGD,
-    'sgd_experimental': tf.keras.optimizers.experimental.SGD,
+    # TODO(chenmoneygithub): experimental.SGD
    'adam': tf.keras.optimizers.Adam,
-    'adam_experimental': tf.keras.optimizers.experimental.Adam,
+    # TODO(chenmoneygithub): experimental.Adam
    'adamw': nlp_optimization.AdamWeightDecay,
    'lamb': tfa_optimizers.LAMB,
    'rmsprop': tf.keras.optimizers.RMSprop,