@@ -113,6 +113,7 @@ class Optimizers:
 
     SGD = "SGD"
     ADAM = "Adam"
+    ADAMW = "AdamW"
     RMS_PROP = "RMSprop"
     RMS_PROP_TF = "RMSpropTF"
     LAMB = "Lamb"

@@ -7,6 +7,7 @@ from super_gradients.training.utils.optimizers.lamb import Lamb
 OPTIMIZERS = {
     Optimizers.SGD: optim.SGD,
     Optimizers.ADAM: optim.Adam,
+    Optimizers.ADAMW: optim.AdamW,
     Optimizers.RMS_PROP: optim.RMSprop,
     Optimizers.RMS_PROP_TF: RMSpropTF,
     Optimizers.LAMB: Lamb,