Skip to content

Commit 2bf410a

Browse files
Gin-Config Teamcopybara-github
authored andcommitted
Added AdamW from tf.keras.optimizers to external_configurables.
AdamW optimizer is supported in current version of tensorflow(https://www.tensorflow.org/api_docs/python/tf/keras/optimizers) PiperOrigin-RevId: 636486640
1 parent 1507860 commit 2bf410a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

gin/tf/external_configurables.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,8 @@ def piecewise_constant(global_step, *args, **kwargs):
9898
tf.keras.optimizers.Adagrad, module='tf.keras.optimizers')
9999
config.external_configurable(
100100
tf.keras.optimizers.Adam, module='tf.keras.optimizers')
101+
config.external_configurable(
102+
tf.keras.optimizers.AdamW, module='tf.keras.optimizers')
101103
config.external_configurable(
102104
tf.keras.optimizers.Adamax, module='tf.keras.optimizers')
103105
config.external_configurable(

0 commit comments

Comments
 (0)