Skip to content

Commit 217350a

Browse files
committed
Update threshold to fix failing tests.
1 parent 550f05e commit 217350a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def _test_detection_model(self, name, dev):
104104
kwargs = {}
105105
if "retinanet" in name:
106106
# Reduce the default threshold to ensure the returned boxes are not empty.
107-
kwargs["score_thresh"] = 0.0099999
107+
kwargs["score_thresh"] = 0.009
108108
model = models.detection.__dict__[name](num_classes=50, pretrained_backbone=False, **kwargs)
109109
model.eval().to(device=dev)
110110
input_shape = (3, 300, 300)

0 commit comments

Comments
 (0)