Skip to content

Commit 66d861f

Browse files
committed
Change test threshold to resolve flaky test.
1 parent 4c6c112 commit 66d861f

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed
Binary file not shown.

test/test_models.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,9 @@ def _test_detection_model(self, name, dev):
110110
elif "fasterrcnn_mobilenet_v3_large_fpn" in name:
111111
kwargs["box_score_thresh"] = 0.02076
112112
elif "fasterrcnn_mobilenet_v3_large_320_fpn" in name:
113-
kwargs["box_score_thresh"] = 0.0208
113+
kwargs["box_score_thresh"] = 0.02082
114+
kwargs["rpn_pre_nms_top_n_test"] = 1000
115+
kwargs["rpn_post_nms_top_n_test"] = 1000
114116
model = models.detection.__dict__[name](num_classes=50, pretrained_backbone=False, **kwargs)
115117
model.eval().to(device=dev)
116118
input_shape = (3, 300, 300)

0 commit comments

Comments
 (0)