Skip to content

GitLab

Commit c715b138 authored by Jennifer Wei's avatar Jennifer Wei
Browse files

Merge remote-tracking branch 'refs/remotes/jnwei/pl_upgrades' into pl_upgrades

parents e3e09c46 76fb7ce6
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 8 deletions
train_openfold.py
View file @ c715b138
...@@ -682,8 +682,8 @@ if __name__ == "__main__": ...@@ -682,8 +682,8 @@ if __name__ == "__main__":
trainer_group.add_argument( trainer_group.add_argument(
"--reload_dataloaders_every_n_epochs", type=int, default=1, "--reload_dataloaders_every_n_epochs", type=int, default=1,
) )
trainer_group.add_argument(
trainer_group.add_argument("--accumulate_grad_batches", type=int, default=1, "--accumulate_grad_batches", type=int, default=1,
help="Accumulate gradients over k batches before next optimizer step.") help="Accumulate gradients over k batches before next optimizer step.")
args = parser.parse_args() args = parser.parse_args()
...@@ -700,5 +700,4 @@ if __name__ == "__main__": ...@@ -700,5 +700,4 @@ if __name__ == "__main__":
raise ValueError( raise ValueError(
"Choose between loading pretrained Jax-weights and a checkpoint-path") "Choose between loading pretrained Jax-weights and a checkpoint-path")
main(args) main(args)
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