Make new Keras optimizer available in MG's github repo.

PiperOrigin-RevId: 477566690

Make new Keras optimizer available in MG's github repo.
PiperOrigin-RevId: 477566690
89bd20c5 · Chen Qian · A. Unique TensorFlower · 2d3235f8 · 89bd20c5
Commit 89bd20c5 authored Sep 28, 2022 by Chen Qian Committed by A. Unique TensorFlower Sep 28, 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
@@ -29,8 +29,8 @@ from official.modeling.optimization.configs import optimization_config as opt_cf

 # Optimizer CLS to be used in both legacy and new path.
 SHARED_OPTIMIZERS = {
-    # TODO(chenmoneygithub): experimental.SGD
-    # TODO(chenmoneygithub): experimental.Adam
+    'sgd_experimental': tf.keras.optimizers.experimental.SGD,
+    'adam_experimental': tf.keras.optimizers.experimental.Adam,
    'adamw': legacy_adamw.AdamWeightDecay,
    'adamw_experimental': tf.keras.optimizers.experimental.AdamW,
    'lamb': tfa_optimizers.LAMB,