fix yolox cuda unittest (open-mmlab#6636)

ZwwWayne · Jul 18, 2022 · c504a08 · c504a08
1 parent 243f39b
commit c504a08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models/test_forward.py b/tests/test_models/test_forward.py
@@ -704,4 +704,4 @@ def test_yolox_random_size():
         gt_bboxes=gt_bboxes,
         gt_labels=gt_labels,
         return_loss=True)
-    assert detector.input_size == (64, 64)
+    assert detector._input_size == (64, 64)