From 7c160e4315b5fdabe9db6babf06c1e5586a53c40 Mon Sep 17 00:00:00 2001
From: secutron <secutron@naver.com>
Date: Thu, 9 May 2024 10:38:41 +0900
Subject: [PATCH] hotfix: add lighting dependancy for _cosine_scheduler_fn(..)

---
 model/marlin.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/model/marlin.py b/model/marlin.py
index e3e2eaf..7b5bf2e 100644
--- a/model/marlin.py
+++ b/model/marlin.py
@@ -307,6 +307,7 @@ def _cosine_scheduler_factors(self):
         return values_factors
 
     def _cosine_scheduler_fn(self, epoch):
+        if epoch > 0: epoch -= 1  # lightning dependancy
         return self.lr_scheduler_factors[epoch]
 
     def configure_optimizers(self):