Skip to content

GitLab

Unverified Commit ca43b8dd authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Reformat train_openfold.py

parent 9ce96fb5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
train_openfold.py
View file @ ca43b8dd
...@@ -153,7 +153,8 @@ def main(args): ...@@ -153,7 +153,8 @@ def main(args):
strict=True, strict=True,
) )
callbacks.append(es) callbacks.append(es)
if args.log_performance:
if(args.log_performance):
global_batch_size = args.num_nodes * args.gpus global_batch_size = args.num_nodes * args.gpus
perf = PerformanceLoggingCallback( perf = PerformanceLoggingCallback(
log_file=os.path.join(args.output_dir, "performance_log.json"), log_file=os.path.join(args.output_dir, "performance_log.json"),
......
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