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

train_openfold.py

@@ -682,8 +682,8 @@ if __name__ == "__main__":
     trainer_group.add_argument(
         "--reload_dataloaders_every_n_epochs", type=int, default=1,
     )
-
-    trainer_group.add_argument("--accumulate_grad_batches", type=int, default=1,
+    trainer_group.add_argument(
+        "--accumulate_grad_batches", type=int, default=1,
         help="Accumulate gradients over k batches before next optimizer step.")
 
     args = parser.parse_args()
@@ -700,5 +700,4 @@ if __name__ == "__main__":
         raise ValueError(
             "Choose between loading pretrained Jax-weights and a checkpoint-path")
 
-
     main(args)