From d8377384a2c2854b6892a110b7fecfb93720b057 Mon Sep 17 00:00:00 2001
From: 5thGenDev <44685200+5thGenDev@users.noreply.github.com>
Date: Sun, 6 Aug 2023 19:31:34 +0100
Subject: [PATCH] Update lr_schedulers.py

---
 src/lr_schedulers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lr_schedulers.py b/src/lr_schedulers.py
index 894fbee..583ba4c 100644
--- a/src/lr_schedulers.py
+++ b/src/lr_schedulers.py
@@ -6,9 +6,9 @@
 def init_lr_scheduler(
     optimizer,
     lr_scheduler,  # learning rate scheduler
+    epochs,
     stepsize=[20, 40],  # step size to decay learning rate
     gamma=0.1,  # learning rate decay
-    epochs=600,
 ):
     '''
     Read this line of code: https://github.com/pytorch/vision/blob/main/references/classification/train.py#L304