From 9cefc71a58ef5a46f7d66eff2b6f2226e12d5e33 Mon Sep 17 00:00:00 2001
From: Riko Uphoff <riko.uphoff@student.uni-halle.de>
Date: Fri, 28 Mar 2025 17:35:55 +0100
Subject: [PATCH] Fixed naming bug
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index 7342f3d..2cc3203 100644
--- a/main.py
+++ b/main.py
@@ -74,7 +74,7 @@ if __name__ == "__main__":
optimizer, model = get_optimizer(args, model)
- num_steps = ceil(args.epochs * len(dataset) / args.batch_size)
+ num_steps = ceil(args.num_epochs * len(dataset) / args.batch_size)
scheduler = get_scheduler(
optimizer, args.lr_scheduler, args.warm_up_fraction, num_steps, args.lr, args.lr_min
)
--
GitLab