Update GANDLF/optimizers/wrap_monai.py

mlcommons · Jan 26, 2024 · 0528a03 · 0528a03
1 parent d3a5d42
commit 0528a03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/optimizers/wrap_monai.py b/GANDLF/optimizers/wrap_monai.py
@@ -10,4 +10,4 @@ def novograd_wrapper(parameters):
         eps=parameters["optimizer"].get("eps", 1e-8),
         weight_decay=parameters["optimizer"].get("weight_decay", 3e-05),
         amsgrad=parameters["optimizer"].get("amsgrad", False),
-    )
+    )