add the missing resume_last_lr_step()

train_openfold.py

@@ -244,6 +244,9 @@ class OpenFoldWrapper(pl.LightningModule):
     def on_save_checkpoint(self, checkpoint):
         checkpoint["ema"] = self.ema.state_dict()
 
+    def resume_last_lr_step(self, lr_step):
+        self.last_lr_step = lr_step
+
 
 def main(args):
     if(args.seed is not None):