From 4584bc3bfd1d4ec8ad50222e40097d2c76c83ca8 Mon Sep 17 00:00:00 2001 From: Pavle Josipovic Date: Thu, 28 Nov 2024 07:45:56 +0000 Subject: [PATCH] Increase perf margin on unet In order to make device perf pipeline more stable, bump perf margin on unet device perf from 1% to 2% --- models/experimental/functional_unet/tests/test_unet_perf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/experimental/functional_unet/tests/test_unet_perf.py b/models/experimental/functional_unet/tests/test_unet_perf.py index 3d22980550b..0f44c7d7554 100644 --- a/models/experimental/functional_unet/tests/test_unet_perf.py +++ b/models/experimental/functional_unet/tests/test_unet_perf.py @@ -48,7 +48,7 @@ def test_unet_perf_device(batch: int, groups: int, expected_device_perf_fps: flo ) expected_perf_cols = {inference_time_key: expected_device_perf_fps} expected_results = check_device_perf( - post_processed_results, margin=0.01, expected_perf_cols=expected_perf_cols, assert_on_fail=True + post_processed_results, margin=0.02, expected_perf_cols=expected_perf_cols, assert_on_fail=True ) prep_device_perf_report( model_name=f"unet-shallow_batch-{batch}_groups-{groups}",