bench try

openvinotoolkit · Jul 17, 2024 · b0264e8 · b0264e8
1 parent d73afa0
commit b0264e8
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/tests/post_training/pipelines/base.py b/tests/post_training/pipelines/base.py
@@ -409,14 +409,18 @@ def run_bench(self) -> None:
         """
         if not self.run_benchmark_app:
             return
-        runner = Command(f"benchmark_app -m {self.path_compressed_ir}")
-        runner.run(stdout=False)
-        cmd_output = " ".join(runner.output)
-
-        match = re.search(r"Throughput\: (.+?) FPS", cmd_output)
-        if match is not None:
-            fps = match.group(1)
-            self.run_info.fps = float(fps)
+
+        try:
+            runner = Command(f"benchmark_app -m {self.path_compressed_ir}")
+            runner.run(stdout=False)
+            cmd_output = " ".join(runner.output)
+
+            match = re.search(r"Throughput\: (.+?) FPS", cmd_output)
+            if match is not None:
+                fps = match.group(1)
+                self.run_info.fps = float(fps)
+        except Exception as e:
+            print(e)
 
     def cleanup_cache(self):
         """