yolo_box_test_time_lower (#62368)

PaddlePaddle · Mar 4, 2024 · bdd1fe8 · bdd1fe8
1 parent 14b3c61
commit bdd1fe8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test/ir/inference/test_trt_convert_yolo_box.py b/test/ir/inference/test_trt_convert_yolo_box.py
@@ -56,13 +56,13 @@ def generate_input2(attrs: list[dict[str, Any]], batch):
             iou_aware,
             iou_aware_factor,
         ) in product(
-            [1, 4],
-            [80, 30],
+            [1],
+            [80],
             [[10, 13, 16, 30, 33, 23]],
-            [32, 16],
-            [0.01, 0.02],
+            [32],
+            [0.01],
             [True, False],
-            [1.0, 0.9],
+            [1.0],
             [False, True],
             [0.5],
         ):