Update threshold to fix failing tests.

pytorch · Jan 6, 2021 · 217350a · 217350a
1 parent 550f05e
commit 217350a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_models.py b/test/test_models.py
@@ -104,7 +104,7 @@ def _test_detection_model(self, name, dev):
         kwargs = {}
         if "retinanet" in name:
             # Reduce the default threshold to ensure the returned boxes are not empty.
-            kwargs["score_thresh"] = 0.0099999
+            kwargs["score_thresh"] = 0.009
         model = models.detection.__dict__[name](num_classes=50, pretrained_backbone=False, **kwargs)
         model.eval().to(device=dev)
         input_shape = (3, 300, 300)