pass callbacks to trainer

005e6125 · Marta · 6c14c76a · 005e6125
Commit 005e6125 authored Nov 02, 2021 by Marta
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

train_openfold.py train_openfold.py +1 -0

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -167,6 +167,7 @@ def main(args):
    trainer = pl.Trainer.from_argparse_args(
        args,
        strategy=strategy,
+        callbacks=callbacks,
    )

    if(args.resume_model_weights_only):