deferred benchmark metrics logging

huggingface · Feb 6, 2024 · 67800e3 · 67800e3
1 parent 11d2573
commit 67800e3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/optimum_benchmark/cli.py b/optimum_benchmark/cli.py
@@ -76,4 +76,8 @@ def benchmark_cli(experiment_config: DictConfig) -> None:
 
     benchmark_report = launch(experiment_config=experiment_config)
 
+    LOGGER.info("Benchmark Report:")
+    for metric, value in benchmark_report.items():
+        LOGGER.info(f"\t+  {metric}: {value:.3f}")
+
     json.dump(benchmark_report, open("benchmark_report.json", "w"), indent=4)