Skip to content

GitLab

Commit f85d67f4 authored by sachinkadyan7's avatar sachinkadyan7 Committed by Sachin Kadyan
Browse files

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

parent 9b114f28
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
openfold/utils/script_utils.py
View file @ f85d67f4
...@@ -159,7 +159,7 @@ def run_model(model, batch, tag, output_dir): ...@@ -159,7 +159,7 @@ def run_model(model, batch, tag, output_dir):
out = model(batch) out = model(batch)
inference_time = time.perf_counter() - t inference_time = time.perf_counter() - t
logger.info(f"Inference time: {inference_time}") 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 model.config.template.enabled = template_enabled
......
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