From 2bf410a6a724c24e3260d34f7df6a8fa3b7b41ea Mon Sep 17 00:00:00 2001 From: Gin-Config Team Date: Thu, 23 May 2024 03:25:26 -0700 Subject: [PATCH] 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 --- gin/tf/external_configurables.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gin/tf/external_configurables.py b/gin/tf/external_configurables.py index 8c21eb5..f52f3c8 100644 --- a/gin/tf/external_configurables.py +++ b/gin/tf/external_configurables.py @@ -98,6 +98,8 @@ def piecewise_constant(global_step, *args, **kwargs): tf.keras.optimizers.Adagrad, module='tf.keras.optimizers') config.external_configurable( tf.keras.optimizers.Adam, module='tf.keras.optimizers') +config.external_configurable( + tf.keras.optimizers.AdamW, module='tf.keras.optimizers') config.external_configurable( tf.keras.optimizers.Adamax, module='tf.keras.optimizers') config.external_configurable(