Fix bug in lr scheduler configure

luigibonati · Jan 12, 2024 · 911f3ef · 911f3ef
1 parent 04596c2
commit 911f3ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlcolvar/cvs/cv.py b/mlcolvar/cvs/cv.py
@@ -206,7 +206,7 @@ def configure_optimizers(self):
         if self.lr_scheduler_kwargs:
             scheduler_cls = self.lr_scheduler_kwargs['scheduler']
             scheduler_kwargs = {k: v for k, v in self.lr_scheduler_kwargs.items() if k != 'scheduler'}
-            lr_scheduler = scheduler_cls(**scheduler_kwargs)
+            lr_scheduler = scheduler_cls(optimizer, **scheduler_kwargs)
             return [optimizer] , [lr_scheduler]
         else: 
             return optimizer