From c9fd0ec4b39e4260a44d8889570f20a3c24b12a5 Mon Sep 17 00:00:00 2001 From: "PJLAB\\huanghaian" <1286304229@qq.com> Date: Tue, 30 Nov 2021 13:12:02 +0800 Subject: [PATCH] fix yolox cuda unittest --- tests/test_models/test_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models/test_forward.py b/tests/test_models/test_forward.py index 0730b244643..c0611be9996 100644 --- 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)