fix number of iterations (#1509)

dsikka · web-flow · commit 317ef355a644 · 2024-01-05T10:20:16.000-05:00
diff --git a/src/deepsparse/benchmark/benchmark_pipeline.py b/src/deepsparse/benchmark/benchmark_pipeline.py
@@ -511,7 +511,7 @@ def main(
     benchmark_results = {
         "items_per_sec": items_per_sec,
         "seconds_ran": total_run_time,
-        "iterations": len(batch_times),
+        "iterations": len(batch_times["total_inference"]),
         "compute_sections": section_stats,
     }
 

Original file line number	Diff line number	Diff line change
`@@ -511,7 +511,7 @@ def main(`
`511`	`511`	`benchmark_results = {`
`512`	`512`	`"items_per_sec": items_per_sec,`
`513`	`513`	`"seconds_ran": total_run_time,`
`514`		`- "iterations": len(batch_times),`
	`514`	`+ "iterations": len(batch_times["total_inference"]),`
`515`	`515`	`"compute_sections": section_stats,`
`516`	`516`	`}`
`517`	`517`