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):