Skip to content

Commit ad71d79

Browse files
authored
Update test_inference.py
1 parent 999a0c5 commit ad71d79

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_inference.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ def test_test_time_augmentation_on_cpu():
99
config_file = 'configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py'
1010
config = mmcv.Config.fromfile(config_file)
1111

12+
# Remove pretrain model download for testing
13+
config.model.pretrained = None
1214
# Replace SyncBN with BN to inference on CPU
1315
norm_cfg = dict(type='BN', requires_grad=True)
1416
config.model.backbone.norm_cfg = norm_cfg

0 commit comments

Comments
 (0)