Bug with lightning LR scheduler fixed

BojarLab · Sep 11, 2024 · d11fac5 · d11fac5
1 parent 28dcaed
commit d11fac5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gifflar/model/pretrain.py b/gifflar/model/pretrain.py
@@ -208,3 +208,6 @@ def configure_optimizers(self):
             "monitor": "val/loss",
         }
 
+    def lr_scheduler_step(self, scheduler, metric) -> None:
+        scheduler.step(self.current_epoch)
+