Reformat train_openfold.py

ca43b8dd · Gustaf Ahdritz · GitHub · 9ce96fb5 · ca43b8dd
Unverified Commit ca43b8dd authored Dec 28, 2021 by Gustaf Ahdritz Committed by GitHub Dec 28, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

train_openfold.py train_openfold.py +2 -1

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -153,7 +153,8 @@ def main(args):
            strict=True,
        )
        callbacks.append(es)
-    if args.log_performance:
+        
+    if(args.log_performance):
        global_batch_size = args.num_nodes * args.gpus
        perf = PerformanceLoggingCallback(
            log_file=os.path.join(args.output_dir, "performance_log.json"),