From d11fac5d9c3d31a2f47bb5c77c90dd2aebdb617f Mon Sep 17 00:00:00 2001
From: Roman Joeres <Roman.Joeres@helmholtz-hzi.de>
Date: Wed, 11 Sep 2024 11:27:34 +0200
Subject: [PATCH] Bug with lightning LR scheduler fixed

---
 gifflar/model/pretrain.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gifflar/model/pretrain.py b/gifflar/model/pretrain.py
index 7da35a9..8240507 100644
--- 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)
+