From 9887db0957b6acf43b753c718216afbccd7eaa08 Mon Sep 17 00:00:00 2001
From: 5thGenDev <44685200+5thGenDev@users.noreply.github.com>
Date: Sun, 6 Aug 2023 19:30:51 +0100
Subject: [PATCH] Update args.py

---
 args.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/args.py b/args.py
index fb38a8a..866a3a0 100644
--- 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,
     }