Skip to content

GitLab

Commit af788dab authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Remove experimental optimizer usage on github. 

PiperOrigin-RevId: 435102510
parent 0f9f7c8e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/modeling/optimization/optimizer_factory.py
View file @ af788dab
...@@ -28,9 +28,9 @@ from official.nlp import optimization as nlp_optimization ...@@ -28,9 +28,9 @@ from official.nlp import optimization as nlp_optimization
OPTIMIZERS_CLS = { OPTIMIZERS_CLS = {
'sgd': tf.keras.optimizers.SGD, 'sgd': tf.keras.optimizers.SGD,
'sgd_experimental': tf.keras.optimizers.experimental.SGD, # TODO(chenmoneygithub): experimental.SGD
'adam': tf.keras.optimizers.Adam, 'adam': tf.keras.optimizers.Adam,
'adam_experimental': tf.keras.optimizers.experimental.Adam, # TODO(chenmoneygithub): experimental.Adam
'adamw': nlp_optimization.AdamWeightDecay, 'adamw': nlp_optimization.AdamWeightDecay,
'lamb': tfa_optimizers.LAMB, 'lamb': tfa_optimizers.LAMB,
'rmsprop': tf.keras.optimizers.RMSprop, 'rmsprop': tf.keras.optimizers.RMSprop,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment