Skip to content

GitLab

Commit 89bd20c5 authored by Chen Qian's avatar Chen Qian Committed by A. Unique TensorFlower
Browse files

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

PiperOrigin-RevId: 477566690
parent 2d3235f8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/modeling/optimization/optimizer_factory.py
View file @ 89bd20c5
...@@ -29,8 +29,8 @@ from official.modeling.optimization.configs import optimization_config as opt_cf ...@@ -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. # Optimizer CLS to be used in both legacy and new path.
SHARED_OPTIMIZERS = { SHARED_OPTIMIZERS = {
# TODO(chenmoneygithub): experimental.SGD 'sgd_experimental': tf.keras.optimizers.experimental.SGD,
# TODO(chenmoneygithub): experimental.Adam 'adam_experimental': tf.keras.optimizers.experimental.Adam,
'adamw': legacy_adamw.AdamWeightDecay, 'adamw': legacy_adamw.AdamWeightDecay,
'adamw_experimental': tf.keras.optimizers.experimental.AdamW, 'adamw_experimental': tf.keras.optimizers.experimental.AdamW,
'lamb': tfa_optimizers.LAMB, 'lamb': tfa_optimizers.LAMB,
......
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