diff --git a/tests/test_models/test_backbone.py b/tests/test_models/test_backbone.py index 36fdd78fc..084849aa4 100644 --- a/tests/test_models/test_backbone.py +++ b/tests/test_models/test_backbone.py @@ -676,13 +676,13 @@ def test_fastscnn_backbone(): model.init_weights() model.train() batch_size = 4 - imgs = torch.randn(batch_size, 3, 1024, 2048) + imgs = torch.randn(batch_size, 3, 512, 1024) feat = model(imgs) assert len(feat) == 3 # higher-res - assert feat[0].shape == torch.Size([batch_size, 64, 128, 256]) + assert feat[0].shape == torch.Size([batch_size, 64, 64, 128]) # lower-res - assert feat[1].shape == torch.Size([batch_size, 128, 32, 64]) + assert feat[1].shape == torch.Size([batch_size, 128, 16, 32]) # FFM output - assert feat[2].shape == torch.Size([batch_size, 128, 128, 256]) + assert feat[2].shape == torch.Size([batch_size, 128, 64, 128]) diff --git a/tests/test_ops/test_inverted_residual_module.py b/tests/test_ops/test_inverted_residual_module.py index 0e3d876e3..41c663b99 100644 --- a/tests/test_ops/test_inverted_residual_module.py +++ b/tests/test_ops/test_inverted_residual_module.py @@ -10,7 +10,7 @@ def test_inv_residual(): # test stride assertion. InvertedResidual(32, 32, 3, 4) - # test default config without res connection. + # test default config with res connection. # set expand_ratio = 4, stride = 1 and inp=oup. inv_module = InvertedResidual(32, 32, 1, 4) assert inv_module.use_res_connect