Support any lr_scheduler

2026-09-12 12:40:20 +08:00 · 2019-07-26 11:03:44 +07:00
parent c4b37d1efe
commit 7d97e3e6e4
9 changed files with 20 additions and 41 deletions
@@ -58,7 +58,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):

    def configure_optimizers(self):
        """
-        Return array of optimizers
+        Return a list of optimizers and a list of schedulers (could be empty)
        :return:
        """
        raise NotImplementedError