We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 999a0c5 commit ad71d79Copy full SHA for ad71d79
tests/test_inference.py
@@ -9,6 +9,8 @@ def test_test_time_augmentation_on_cpu():
9
config_file = 'configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py'
10
config = mmcv.Config.fromfile(config_file)
11
12
+ # Remove pretrain model download for testing
13
+ config.model.pretrained = None
14
# Replace SyncBN with BN to inference on CPU
15
norm_cfg = dict(type='BN', requires_grad=True)
16
config.model.backbone.norm_cfg = norm_cfg
0 commit comments