fix yolox cuda unittest (#6636)

open-mmlab · Nov 30, 2021 · 73d384f · 73d384f
1 parent b4e30a3
commit 73d384f
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)