Skip to content

GitLab

Commit 69a8a287 authored by Zhang690683220's avatar Zhang690683220
Browse files

add the missing resume_last_lr_step()

parent a7274ef0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
train_openfold.py
View file @ 69a8a287
...@@ -244,6 +244,9 @@ class OpenFoldWrapper(pl.LightningModule): ...@@ -244,6 +244,9 @@ class OpenFoldWrapper(pl.LightningModule):
def on_save_checkpoint(self, checkpoint): def on_save_checkpoint(self, checkpoint):
checkpoint["ema"] = self.ema.state_dict() checkpoint["ema"] = self.ema.state_dict()
def resume_last_lr_step(self, lr_step):
self.last_lr_step = lr_step
def main(args): def main(args):
if(args.seed is not None): if(args.seed is not None):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment