update test tpn

open-mmlab · Apr 27, 2023 · 2110556 · 2110556
1 parent 1918e23
commit 2110556
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/recognizers/test_recognizer2d.py b/tests/models/recognizers/test_recognizer2d.py
@@ -147,7 +147,7 @@ def test_tpn():
         'tpn/tpn-tsm_imagenet-pretrained-r50_8xb8-1x1x8-150e_sthv1-rgb.py')
     config.model['backbone']['pretrained'] = None
 
-    input_shape = (2, 8, 3, 32, 32)
+    input_shape = (1, 8, 3, 64, 64)
     train_test_step(config, input_shape)