未验证 提交 cb81826a 编写于 作者: P Pei Yang 提交者: GitHub

extend multiclass_nms unittest timeout threshold (#32214)

* extend multiclass_nms unittest timeout threshold

* adjust timeout to 200s

* temporarily disable multiclass_nms trt op teller
上级 b9e543f8
...@@ -114,7 +114,6 @@ struct SimpleOpTypeSetTeller : public Teller { ...@@ -114,7 +114,6 @@ struct SimpleOpTypeSetTeller : public Teller {
"yolo_box", "yolo_box",
"roi_align", "roi_align",
"affine_channel", "affine_channel",
"multiclass_nms",
"nearest_interp", "nearest_interp",
"anchor_generator", "anchor_generator",
}; };
......
...@@ -32,4 +32,5 @@ if(WITH_GPU AND TENSORRT_FOUND) ...@@ -32,4 +32,5 @@ if(WITH_GPU AND TENSORRT_FOUND)
set_tests_properties(test_trt_subgraph_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_subgraph_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_trt_activation_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_activation_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_trt_conv_pass PROPERTIES TIMEOUT 120) set_tests_properties(test_trt_conv_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_trt_multiclass_nms_op PROPERTIES TIMEOUT 200)
endif() endif()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册