From b2482eaebc03c0bf50de2b5e6e7d8f169b710759 Mon Sep 17 00:00:00 2001 From: Ainesh Nanda <67788476+Ainesh06102004@users.noreply.github.com> Date: Tue, 12 Mar 2024 21:32:33 +0530 Subject: [PATCH] Update GANDLF/schedulers/wrap_monai.py Co-authored-by: Sarthak Pati --- GANDLF/schedulers/wrap_monai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GANDLF/schedulers/wrap_monai.py b/GANDLF/schedulers/wrap_monai.py index dab01a764..65943a1df 100644 --- 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"], )