Validate that deepsparse.benchmark has recorded samples (#275)

* Validate that deepsparse.benchmark has recorded samples * quality
neuralmagic · Mar 1, 2022 · 16c00dd · 16c00dd
1 parent 3d0d5a0
commit 16c00dd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/deepsparse/benchmark_model/stream_benchmark.py b/src/deepsparse/benchmark_model/stream_benchmark.py
@@ -115,6 +115,11 @@ def model_stream_benchmark(
     else:
         raise Exception(f"Unknown scenario '{scenario}'")
 
+    if len(batch_times) == 0:
+        raise Exception(
+            "Generated no batch timings, try extending benchmark time with '--time'"
+        )
+
     # Convert times to milliseconds
     batch_times_ms = [
         (batch_time[1] - batch_time[0]) * 1000 for batch_time in batch_times