Update GANDLF/schedulers/wrap_monai.py

Co-authored-by: Sarthak Pati <patis@iu.edu>
mlcommons · Mar 12, 2024 · b2482ea · b2482ea
1 parent cce37f7
commit b2482ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/schedulers/wrap_monai.py b/GANDLF/schedulers/wrap_monai.py
@@ -2,7 +2,7 @@
 
 def warmupcosineschedule(parameters):
     return WCS(
-        parameters["optimizer_object"], #optimizers["model_optimizers"],
+        parameters["optimizer_object"],
         t_total = parameters["num_epochs"],
         warmup_steps = 0.1 * parameters["num_epochs"],
     )