Update args.py

5thGenDev · Aug 6, 2023 · 9887db0 · 9887db0
1 parent c6e5c16
commit 9887db0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/args.py b/args.py
@@ -306,7 +306,7 @@ def lr_scheduler_kwargs(parsed_args):
     """
     return {
         "lr_scheduler": parsed_args.lr_scheduler,
+        "epochs": parsed_args.max_epoch,
         "stepsize": parsed_args.stepsize,
         "gamma": parsed_args.gamma,
-        "epochs": parsed_args.max_epoch,
     }