Skip to content

GitLab

Commit 005e6125 authored by Marta's avatar Marta
Browse files

pass callbacks to trainer

parent 6c14c76a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
train_openfold.py
View file @ 005e6125
...@@ -167,6 +167,7 @@ def main(args): ...@@ -167,6 +167,7 @@ def main(args):
trainer = pl.Trainer.from_argparse_args( trainer = pl.Trainer.from_argparse_args(
args, args,
strategy=strategy, strategy=strategy,
callbacks=callbacks,
) )
if(args.resume_model_weights_only): if(args.resume_model_weights_only):
......
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