Fix yolox ci failed (open-mmlab#6864)

ZwwWayne · Jul 19, 2022 · 4f6a787 · 4f6a787
1 parent 2724f05
commit 4f6a787
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/tests/test_models/test_forward.py b/tests/test_models/test_forward.py
@@ -698,10 +698,4 @@ def test_yolox_random_size():
         gt_bboxes=gt_bboxes,
         gt_labels=gt_labels,
         return_loss=True)
-    detector.forward(
-        imgs,
-        img_metas,
-        gt_bboxes=gt_bboxes,
-        gt_labels=gt_labels,
-        return_loss=True)
-    assert detector._input_size == (64, 64)
+    assert detector._input_size == (64, 96)