Add reference to specific issue

e6263425 · Gustaf Ahdritz · 950b02f0 · e6263425
Commit e6263425 authored Feb 22, 2022 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train_openfold.py train_openfold.py +1 -1

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -117,7 +117,7 @@ class OpenFoldWrapper(pl.LightningModule):
        self.ema.update(self.model)

 #    def training_step_end(self, outputs):
-#        # Temporary measure to address DeepSpeed scheduler bug
+#        # Temporary measure to address DeepSpeed scheduler bug (PL issue 11694)
 #        if(self.trainer.global_step != self.last_lr_step):
 #            self.lr_schedulers().step()
 #            self.last_lr_step = self.trainer.global_step