scnc
/
openfold
mirror of https://github.com/aqlaboratory/openfold.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Bugfix for timings.json - now store timings per tag.

This commit is contained in:
sachinkadyan7 2023-09-13 13:19:31 -07:00 committed by Sachin Kadyan
parent 9b114f28df
commit f85d67f4f9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
openfold/utils/script_utils.py
View File

@ -159,7 +159,7 @@ def run_model(model, batch, tag, output_dir):
out = model(batch)
inference_time = time.perf_counter() - t
logger.info(f"Inference time: {inference_time}")
update_timings({"inference": inference_time}, os.path.join(output_dir, "timings.json"))
update_timings({tag: {"inference": inference_time}}, os.path.join(output_dir, "timings.json"))
model.config.template.enabled = template_enabled