Skip to content

GitLab

Commit 226c7bac authored by Jennifer's avatar Jennifer
Browse files

DNS: Local profiler changes

parent eba00d85
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
train_openfold.py
View file @ 226c7bac
...@@ -435,6 +435,7 @@ def main(args): ...@@ -435,6 +435,7 @@ def main(args):
strategy=strategy, strategy=strategy,
callbacks=callbacks, callbacks=callbacks,
logger=loggers, logger=loggers,
profiler='simple',
) )
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