[stdc2] missing keys in source state_dict

[jeonjungho]

i find the same issue --> #1303 , #1418

but don't solve them

i downloaded pretrained weight stdc2_20220308-7dbd9127.pth from 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/stdc/stdc2_20220308-7dbd9127.pth'

run the code below part

error

1. model and loaded state dict do not match exactly

code
##########################################################
config_file = 'configs/stdc/stdc2_in1k-pre_512x1024_80k_cityscapes.py'
model = init_segmentor(config_file, device='cuda:0')

cfg = Config.fromfile('configs/stdc/stdc2_in1k-pre_512x1024_80k_cityscapes.py')
cfg.load_from = 'pretrained/stdc2_20220308-7dbd9127.pth'

model = build_segmentor(cfg.model)
##########################################################

error traceback
##########################################################
the model and loaded state dict do not match exactly

unexpected key in source state_dict: stages.0.conv.weight, stages.0.bn.weight, stages.0.bn.bias, stages.0.bn.running_mean, stages.0.bn.running_var, stages.0.bn.num_batches_tracked, stages.1.conv.weight, stages.1.bn.weight, stages.1.bn.bias, stages.1.bn.running_mean, stages.1.bn.running_var, stages.1.bn.num_batches_tracked, stages.2.0.layers.0.conv.weight, stages.2.0.layers.0.bn.weight, stages.2.0.layers.0.bn.bias, stages.2.0.layers.0.bn.running_mean, stages.2.0.layers.0.bn.running_var, stages.2.0.layers.0.bn.num_batches_tracked, stages.2.0.layers.1.conv.weight, stages.2.0.layers.1.bn.weight, stages.2.0.layers.1.bn.bias, stages.2.0.layers.1.bn.running_mean, stages.2.0.layers.1.bn.running_var, stages.2.0.layers.1.bn.num_batches_tracked, stages.2.0.layers.2.conv.weight, stages.2.0.layers.2.bn.weight, stages.2.0.layers.2.bn.bias, stages.2.0.layers.2.bn.running_mean, stages.2.0.layers.2.bn.running_var, stages.2.0.layers.2.bn.num_batches_tracked, stages.2.0.layers.3.conv.weight, stages.2.0.layers.3.bn.weight, stages.2.0.layers.3.bn.bias, stages.2.0.layers.3.bn.running_mean, stages.2.0.layers.3.bn.running_var, stages.2.0.layers.3.bn.num_batches_tracked, stages.2.0.downsample.conv.weight, stages.2.0.downsample.bn.weight, stages.2.0.downsample.bn.bias, stages.2.0.downsample.bn.running_mean, stages.2.0.downsample.bn.running_var, stages.2.0.downsample.bn.num_batches_tracked, stages.2.1.layers.0.conv.weight, stages.2.1.layers.0.bn.weight, stages.2.1.layers.0.bn.bias, stages.2.1.layers.0.bn.running_mean, stages.2.1.layers.0.bn.running_var, stages.2.1.layers.0.bn.num_batches_tracked, stages.2.1.layers.1.conv.weight, stages.2.1.layers.1.bn.weight, stages.2.1.layers.1.bn.bias, stages.2.1.layers.1.bn.running_mean, stages.2.1.layers.1.bn.running_var, stages.2.1.layers.1.bn.num_batches_tracked, stages.2.1.layers.2.conv.weight, stages.2.1.layers.2.bn.weight, stages.2.1.layers.2.bn.bias, stages.2.1.layers.2.bn.running_mean, stages.2.1.layers.2.bn.running_var, stages.2.1.layers.2.bn.num_batches_tracked, stages.2.1.layers.3.conv.weight, stages.2.1.layers.3.bn.weight, stages.2.1.layers.3.bn.bias, stages.2.1.layers.3.bn.running_mean, stages.2.1.layers.3.bn.running_var, stages.2.1.layers.3.bn.num_batches_tracked, stages.2.2.layers.0.conv.weight, stages.2.2.layers.0.bn.weight, stages.2.2.layers.0.bn.bias, stages.2.2.layers.0.bn.running_mean, stages.2.2.layers.0.bn.running_var, stages.2.2.layers.0.bn.num_batches_tracked, stages.2.2.layers.1.conv.weight, stages.2.2.layers.1.bn.weight, stages.2.2.layers.1.bn.bias, stages.2.2.layers.1.bn.running_mean, stages.2.2.layers.1.bn.running_var, stages.2.2.layers.1.bn.num_batches_tracked, stages.2.2.layers.2.conv.weight, stages.2.2.layers.2.bn.weight, stages.2.2.layers.2.bn.bias, stages.2.2.layers.2.bn.running_mean, stages.2.2.layers.2.bn.running_var, stages.2.2.layers.2.bn.num_batches_tracked, stages.2.2.layers.3.conv.weight, stages.2.2.layers.3.bn.weight, stages.2.2.layers.3.bn.bias, stages.2.2.layers.3.bn.running_mean, stages.2.2.layers.3.bn.running_var, stages.2.2.layers.3.bn.num_batches_tracked, stages.2.3.layers.0.conv.weight, stages.2.3.layers.0.bn.weight, stages.2.3.layers.0.bn.bias, stages.2.3.layers.0.bn.running_mean, stages.2.3.layers.0.bn.running_var, stages.2.3.layers.0.bn.num_batches_tracked, stages.2.3.layers.1.conv.weight, stages.2.3.layers.1.bn.weight, stages.2.3.layers.1.bn.bias, stages.2.3.layers.1.bn.running_mean, stages.2.3.layers.1.bn.running_var, stages.2.3.layers.1.bn.num_batches_tracked, stages.2.3.layers.2.conv.weight, stages.2.3.layers.2.bn.weight, stages.2.3.layers.2.bn.bias, stages.2.3.layers.2.bn.running_mean, stages.2.3.layers.2.bn.running_var, stages.2.3.layers.2.bn.num_batches_tracked, stages.2.3.layers.3.conv.weight, stages.2.3.layers.3.bn.weight, stages.2.3.layers.3.bn.bias, stages.2.3.layers.3.bn.running_mean, stages.2.3.layers.3.bn.running_var, stages.2.3.layers.3.bn.num_batches_tracked, stages.3.0.layers.0.conv.weight, stages.3.0.layers.0.bn.weight, stages.3.0.layers.0.bn.bias, stages.3.0.layers.0.bn.running_mean, stages.3.0.layers.0.bn.running_var, stages.3.0.layers.0.bn.num_batches_tracked, stages.3.0.layers.1.conv.weight, stages.3.0.layers.1.bn.weight, stages.3.0.layers.1.bn.bias, stages.3.0.layers.1.bn.running_mean, stages.3.0.layers.1.bn.running_var, stages.3.0.layers.1.bn.num_batches_tracked, stages.3.0.layers.2.conv.weight, stages.3.0.layers.2.bn.weight, stages.3.0.layers.2.bn.bias, stages.3.0.layers.2.bn.running_mean, stages.3.0.layers.2.bn.running_var, stages.3.0.layers.2.bn.num_batches_tracked, stages.3.0.layers.3.conv.weight, stages.3.0.layers.3.bn.weight, stages.3.0.layers.3.bn.bias, stages.3.0.layers.3.bn.running_mean, stages.3.0.layers.3.bn.running_var, stages.3.0.layers.3.bn.num_batches_tracked, stages.3.0.downsample.conv.weight, stages.3.0.downsample.bn.weight, stages.3.0.downsample.bn.bias, stages.3.0.downsample.bn.running_mean, stages.3.0.downsample.bn.running_var, stages.3.0.downsample.bn.num_batches_tracked, stages.3.1.layers.0.conv.weight, stages.3.1.layers.0.bn.weight, stages.3.1.layers.0.bn.bias, stages.3.1.layers.0.bn.running_mean, stages.3.1.layers.0.bn.running_var, stages.3.1.layers.0.bn.num_batches_tracked, stages.3.1.layers.1.conv.weight, stages.3.1.layers.1.bn.weight, stages.3.1.layers.1.bn.bias, stages.3.1.layers.1.bn.running_mean, stages.3.1.layers.1.bn.running_var, stages.3.1.layers.1.bn.num_batches_tracked, stages.3.1.layers.2.conv.weight, stages.3.1.layers.2.bn.weight, stages.3.1.layers.2.bn.bias, stages.3.1.layers.2.bn.running_mean, stages.3.1.layers.2.bn.running_var, stages.3.1.layers.2.bn.num_batches_tracked, stages.3.1.layers.3.conv.weight, stages.3.1.layers.3.bn.weight, stages.3.1.layers.3.bn.bias, stages.3.1.layers.3.bn.running_mean, stages.3.1.layers.3.bn.running_var, stages.3.1.layers.3.bn.num_batches_tracked, stages.3.2.layers.0.conv.weight, stages.3.2.layers.0.bn.weight, stages.3.2.layers.0.bn.bias, stages.3.2.layers.0.bn.running_mean, stages.3.2.layers.0.bn.running_var, stages.3.2.layers.0.bn.num_batches_tracked, stages.3.2.layers.1.conv.weight, stages.3.2.layers.1.bn.weight, stages.3.2.layers.1.bn.bias, stages.3.2.layers.1.bn.running_mean, stages.3.2.layers.1.bn.running_var, stages.3.2.layers.1.bn.num_batches_tracked, stages.3.2.layers.2.conv.weight, stages.3.2.layers.2.bn.weight, stages.3.2.layers.2.bn.bias, stages.3.2.layers.2.bn.running_mean, stages.3.2.layers.2.bn.running_var, stages.3.2.layers.2.bn.num_batches_tracked, stages.3.2.layers.3.conv.weight, stages.3.2.layers.3.bn.weight, stages.3.2.layers.3.bn.bias, stages.3.2.layers.3.bn.running_mean, stages.3.2.layers.3.bn.running_var, stages.3.2.layers.3.bn.num_batches_tracked, stages.3.3.layers.0.conv.weight, stages.3.3.layers.0.bn.weight, stages.3.3.layers.0.bn.bias, stages.3.3.layers.0.bn.running_mean, stages.3.3.layers.0.bn.running_var, stages.3.3.layers.0.bn.num_batches_tracked, stages.3.3.layers.1.conv.weight, stages.3.3.layers.1.bn.weight, stages.3.3.layers.1.bn.bias, stages.3.3.layers.1.bn.running_mean, stages.3.3.layers.1.bn.running_var, stages.3.3.layers.1.bn.num_batches_tracked, stages.3.3.layers.2.conv.weight, stages.3.3.layers.2.bn.weight, stages.3.3.layers.2.bn.bias, stages.3.3.layers.2.bn.running_mean, stages.3.3.layers.2.bn.running_var, stages.3.3.layers.2.bn.num_batches_tracked, stages.3.3.layers.3.conv.weight, stages.3.3.layers.3.bn.weight, stages.3.3.layers.3.bn.bias, stages.3.3.layers.3.bn.running_mean, stages.3.3.layers.3.bn.running_var, stages.3.3.layers.3.bn.num_batches_tracked, stages.3.4.layers.0.conv.weight, stages.3.4.layers.0.bn.weight, stages.3.4.layers.0.bn.bias, stages.3.4.layers.0.bn.running_mean, stages.3.4.layers.0.bn.running_var, stages.3.4.layers.0.bn.num_batches_tracked, stages.3.4.layers.1.conv.weight, stages.3.4.layers.1.bn.weight, stages.3.4.layers.1.bn.bias, stages.3.4.layers.1.bn.running_mean, stages.3.4.layers.1.bn.running_var, stages.3.4.layers.1.bn.num_batches_tracked, stages.3.4.layers.2.conv.weight, stages.3.4.layers.2.bn.weight, stages.3.4.layers.2.bn.bias, stages.3.4.layers.2.bn.running_mean, stages.3.4.layers.2.bn.running_var, stages.3.4.layers.2.bn.num_batches_tracked, stages.3.4.layers.3.conv.weight, stages.3.4.layers.3.bn.weight, stages.3.4.layers.3.bn.bias, stages.3.4.layers.3.bn.running_mean, stages.3.4.layers.3.bn.running_var, stages.3.4.layers.3.bn.num_batches_tracked, stages.4.0.layers.0.conv.weight, stages.4.0.layers.0.bn.weight, stages.4.0.layers.0.bn.bias, stages.4.0.layers.0.bn.running_mean, stages.4.0.layers.0.bn.running_var, stages.4.0.layers.0.bn.num_batches_tracked, stages.4.0.layers.1.conv.weight, stages.4.0.layers.1.bn.weight, stages.4.0.layers.1.bn.bias, stages.4.0.layers.1.bn.running_mean, stages.4.0.layers.1.bn.running_var, stages.4.0.layers.1.bn.num_batches_tracked, stages.4.0.layers.2.conv.weight, stages.4.0.layers.2.bn.weight, stages.4.0.layers.2.bn.bias, stages.4.0.layers.2.bn.running_mean, stages.4.0.layers.2.bn.running_var, stages.4.0.layers.2.bn.num_batches_tracked, stages.4.0.layers.3.conv.weight, stages.4.0.layers.3.bn.weight, stages.4.0.layers.3.bn.bias, stages.4.0.layers.3.bn.running_mean, stages.4.0.layers.3.bn.running_var, stages.4.0.layers.3.bn.num_batches_tracked, stages.4.0.downsample.conv.weight, stages.4.0.downsample.bn.weight, stages.4.0.downsample.bn.bias, stages.4.0.downsample.bn.running_mean, stages.4.0.downsample.bn.running_var, stages.4.0.downsample.bn.num_batches_tracked, stages.4.1.layers.0.conv.weight, stages.4.1.layers.0.bn.weight, stages.4.1.layers.0.bn.bias, stages.4.1.layers.0.bn.running_mean, stages.4.1.layers.0.bn.running_var, stages.4.1.layers.0.bn.num_batches_tracked, stages.4.1.layers.1.conv.weight, stages.4.1.layers.1.bn.weight, stages.4.1.layers.1.bn.bias, stages.4.1.layers.1.bn.running_mean, stages.4.1.layers.1.bn.running_var, stages.4.1.layers.1.bn.num_batches_tracked, stages.4.1.layers.2.conv.weight, stages.4.1.layers.2.bn.weight, stages.4.1.layers.2.bn.bias, stages.4.1.layers.2.bn.running_mean, stages.4.1.layers.2.bn.running_var, stages.4.1.layers.2.bn.num_batches_tracked, stages.4.1.layers.3.conv.weight, stages.4.1.layers.3.bn.weight, stages.4.1.layers.3.bn.bias, stages.4.1.layers.3.bn.running_mean, stages.4.1.layers.3.bn.running_var, stages.4.1.layers.3.bn.num_batches_tracked, stages.4.2.layers.0.conv.weight, stages.4.2.layers.0.bn.weight, stages.4.2.layers.0.bn.bias, stages.4.2.layers.0.bn.running_mean, stages.4.2.layers.0.bn.running_var, stages.4.2.layers.0.bn.num_batches_tracked, stages.4.2.layers.1.conv.weight, stages.4.2.layers.1.bn.weight, stages.4.2.layers.1.bn.bias, stages.4.2.layers.1.bn.running_mean, stages.4.2.layers.1.bn.running_var, stages.4.2.layers.1.bn.num_batches_tracked, stages.4.2.layers.2.conv.weight, stages.4.2.layers.2.bn.weight, stages.4.2.layers.2.bn.bias, stages.4.2.layers.2.bn.running_mean, stages.4.2.layers.2.bn.running_var, stages.4.2.layers.2.bn.num_batches_tracked, stages.4.2.layers.3.conv.weight, stages.4.2.layers.3.bn.weight, stages.4.2.layers.3.bn.bias, stages.4.2.layers.3.bn.running_mean, stages.4.2.layers.3.bn.running_var, stages.4.2.layers.3.bn.num_batches_tracked

missing keys in source state_dict: backbone.backbone.stages.0.conv.weight, backbone.backbone.stages.0.bn.weight, backbone.backbone.stages.0.bn.bias, backbone.backbone.stages.0.bn.running_mean, backbone.backbone.stages.0.bn.running_var, backbone.backbone.stages.1.conv.weight, backbone.backbone.stages.1.bn.weight, backbone.backbone.stages.1.bn.bias, backbone.backbone.stages.1.bn.running_mean, backbone.backbone.stages.1.bn.running_var, backbone.backbone.stages.2.0.layers.0.conv.weight, backbone.backbone.stages.2.0.layers.0.bn.weight, backbone.backbone.stages.2.0.layers.0.bn.bias, backbone.backbone.stages.2.0.layers.0.bn.running_mean, backbone.backbone.stages.2.0.layers.0.bn.running_var, backbone.backbone.stages.2.0.layers.1.conv.weight, backbone.backbone.stages.2.0.layers.1.bn.weight, backbone.backbone.stages.2.0.layers.1.bn.bias, backbone.backbone.stages.2.0.layers.1.bn.running_mean, backbone.backbone.stages.2.0.layers.1.bn.running_var, backbone.backbone.stages.2.0.layers.2.conv.weight, backbone.backbone.stages.2.0.layers.2.bn.weight, backbone.backbone.stages.2.0.layers.2.bn.bias, backbone.backbone.stages.2.0.layers.2.bn.running_mean, backbone.backbone.stages.2.0.layers.2.bn.running_var, backbone.backbone.stages.2.0.layers.3.conv.weight, backbone.backbone.stages.2.0.layers.3.bn.weight, backbone.backbone.stages.2.0.layers.3.bn.bias, backbone.backbone.stages.2.0.layers.3.bn.running_mean, backbone.backbone.stages.2.0.layers.3.bn.running_var, backbone.backbone.stages.2.0.downsample.conv.weight, backbone.backbone.stages.2.0.downsample.bn.weight, backbone.backbone.stages.2.0.downsample.bn.bias, backbone.backbone.stages.2.0.downsample.bn.running_mean, backbone.backbone.stages.2.0.downsample.bn.running_var, backbone.backbone.stages.2.1.layers.0.conv.weight, backbone.backbone.stages.2.1.layers.0.bn.weight, backbone.backbone.stages.2.1.layers.0.bn.bias, backbone.backbone.stages.2.1.layers.0.bn.running_mean, backbone.backbone.stages.2.1.layers.0.bn.running_var, backbone.backbone.stages.2.1.layers.1.conv.weight, backbone.backbone.stages.2.1.layers.1.bn.weight, backbone.backbone.stages.2.1.layers.1.bn.bias, backbone.backbone.stages.2.1.layers.1.bn.running_mean, backbone.backbone.stages.2.1.layers.1.bn.running_var, backbone.backbone.stages.2.1.layers.2.conv.weight, backbone.backbone.stages.2.1.layers.2.bn.weight, backbone.backbone.stages.2.1.layers.2.bn.bias, backbone.backbone.stages.2.1.layers.2.bn.running_mean, backbone.backbone.stages.2.1.layers.2.bn.running_var, backbone.backbone.stages.2.1.layers.3.conv.weight, backbone.backbone.stages.2.1.layers.3.bn.weight, backbone.backbone.stages.2.1.layers.3.bn.bias, backbone.backbone.stages.2.1.layers.3.bn.running_mean, backbone.backbone.stages.2.1.layers.3.bn.running_var, backbone.backbone.stages.2.2.layers.0.conv.weight, backbone.backbone.stages.2.2.layers.0.bn.weight, backbone.backbone.stages.2.2.layers.0.bn.bias, backbone.backbone.stages.2.2.layers.0.bn.running_mean, backbone.backbone.stages.2.2.layers.0.bn.running_var, backbone.backbone.stages.2.2.layers.1.conv.weight, backbone.backbone.stages.2.2.layers.1.bn.weight, backbone.backbone.stages.2.2.layers.1.bn.bias, backbone.backbone.stages.2.2.layers.1.bn.running_mean, backbone.backbone.stages.2.2.layers.1.bn.running_var, backbone.backbone.stages.2.2.layers.2.conv.weight, backbone.backbone.stages.2.2.layers.2.bn.weight, backbone.backbone.stages.2.2.layers.2.bn.bias, backbone.backbone.stages.2.2.layers.2.bn.running_mean, backbone.backbone.stages.2.2.layers.2.bn.running_var, backbone.backbone.stages.2.2.layers.3.conv.weight, backbone.backbone.stages.2.2.layers.3.bn.weight, backbone.backbone.stages.2.2.layers.3.bn.bias, backbone.backbone.stages.2.2.layers.3.bn.running_mean, backbone.backbone.stages.2.2.layers.3.bn.running_var, backbone.backbone.stages.2.3.layers.0.conv.weight, backbone.backbone.stages.2.3.layers.0.bn.weight, backbone.backbone.stages.2.3.layers.0.bn.bias, backbone.backbone.stages.2.3.layers.0.bn.running_mean, backbone.backbone.stages.2.3.layers.0.bn.running_var, backbone.backbone.stages.2.3.layers.1.conv.weight, backbone.backbone.stages.2.3.layers.1.bn.weight, backbone.backbone.stages.2.3.layers.1.bn.bias, backbone.backbone.stages.2.3.layers.1.bn.running_mean, backbone.backbone.stages.2.3.layers.1.bn.running_var, backbone.backbone.stages.2.3.layers.2.conv.weight, backbone.backbone.stages.2.3.layers.2.bn.weight, backbone.backbone.stages.2.3.layers.2.bn.bias, backbone.backbone.stages.2.3.layers.2.bn.running_mean, backbone.backbone.stages.2.3.layers.2.bn.running_var, backbone.backbone.stages.2.3.layers.3.conv.weight, backbone.backbone.stages.2.3.layers.3.bn.weight, backbone.backbone.stages.2.3.layers.3.bn.bias, backbone.backbone.stages.2.3.layers.3.bn.running_mean, backbone.backbone.stages.2.3.layers.3.bn.running_var, backbone.backbone.stages.3.0.layers.0.conv.weight, backbone.backbone.stages.3.0.layers.0.bn.weight, backbone.backbone.stages.3.0.layers.0.bn.bias, backbone.backbone.stages.3.0.layers.0.bn.running_mean, backbone.backbone.stages.3.0.layers.0.bn.running_var, backbone.backbone.stages.3.0.layers.1.conv.weight, backbone.backbone.stages.3.0.layers.1.bn.weight, backbone.backbone.stages.3.0.layers.1.bn.bias, backbone.backbone.stages.3.0.layers.1.bn.running_mean, backbone.backbone.stages.3.0.layers.1.bn.running_var, backbone.backbone.stages.3.0.layers.2.conv.weight, backbone.backbone.stages.3.0.layers.2.bn.weight, backbone.backbone.stages.3.0.layers.2.bn.bias, backbone.backbone.stages.3.0.layers.2.bn.running_mean, backbone.backbone.stages.3.0.layers.2.bn.running_var, backbone.backbone.stages.3.0.layers.3.conv.weight, backbone.backbone.stages.3.0.layers.3.bn.weight, backbone.backbone.stages.3.0.layers.3.bn.bias, backbone.backbone.stages.3.0.layers.3.bn.running_mean, backbone.backbone.stages.3.0.layers.3.bn.running_var, backbone.backbone.stages.3.0.downsample.conv.weight, backbone.backbone.stages.3.0.downsample.bn.weight, backbone.backbone.stages.3.0.downsample.bn.bias, backbone.backbone.stages.3.0.downsample.bn.running_mean, backbone.backbone.stages.3.0.downsample.bn.running_var, backbone.backbone.stages.3.1.layers.0.conv.weight, backbone.backbone.stages.3.1.layers.0.bn.weight, backbone.backbone.stages.3.1.layers.0.bn.bias, backbone.backbone.stages.3.1.layers.0.bn.running_mean, backbone.backbone.stages.3.1.layers.0.bn.running_var, backbone.backbone.stages.3.1.layers.1.conv.weight, backbone.backbone.stages.3.1.layers.1.bn.weight, backbone.backbone.stages.3.1.layers.1.bn.bias, backbone.backbone.stages.3.1.layers.1.bn.running_mean, backbone.backbone.stages.3.1.layers.1.bn.running_var, backbone.backbone.stages.3.1.layers.2.conv.weight, backbone.backbone.stages.3.1.layers.2.bn.weight, backbone.backbone.stages.3.1.layers.2.bn.bias, backbone.backbone.stages.3.1.layers.2.bn.running_mean, backbone.backbone.stages.3.1.layers.2.bn.running_var, backbone.backbone.stages.3.1.layers.3.conv.weight, backbone.backbone.stages.3.1.layers.3.bn.weight, backbone.backbone.stages.3.1.layers.3.bn.bias, backbone.backbone.stages.3.1.layers.3.bn.running_mean, backbone.backbone.stages.3.1.layers.3.bn.running_var, backbone.backbone.stages.3.2.layers.0.conv.weight, backbone.backbone.stages.3.2.layers.0.bn.weight, backbone.backbone.stages.3.2.layers.0.bn.bias, backbone.backbone.stages.3.2.layers.0.bn.running_mean, backbone.backbone.stages.3.2.layers.0.bn.running_var, backbone.backbone.stages.3.2.layers.1.conv.weight, backbone.backbone.stages.3.2.layers.1.bn.weight, backbone.backbone.stages.3.2.layers.1.bn.bias, backbone.backbone.stages.3.2.layers.1.bn.running_mean, backbone.backbone.stages.3.2.layers.1.bn.running_var, backbone.backbone.stages.3.2.layers.2.conv.weight, backbone.backbone.stages.3.2.layers.2.bn.weight, backbone.backbone.stages.3.2.layers.2.bn.bias, backbone.backbone.stages.3.2.layers.2.bn.running_mean, backbone.backbone.stages.3.2.layers.2.bn.running_var, backbone.backbone.stages.3.2.layers.3.conv.weight, backbone.backbone.stages.3.2.layers.3.bn.weight, backbone.backbone.stages.3.2.layers.3.bn.bias, backbone.backbone.stages.3.2.layers.3.bn.running_mean, backbone.backbone.stages.3.2.layers.3.bn.running_var, backbone.backbone.stages.3.3.layers.0.conv.weight, backbone.backbone.stages.3.3.layers.0.bn.weight, backbone.backbone.stages.3.3.layers.0.bn.bias, backbone.backbone.stages.3.3.layers.0.bn.running_mean, backbone.backbone.stages.3.3.layers.0.bn.running_var, backbone.backbone.stages.3.3.layers.1.conv.weight, backbone.backbone.stages.3.3.layers.1.bn.weight, backbone.backbone.stages.3.3.layers.1.bn.bias, backbone.backbone.stages.3.3.layers.1.bn.running_mean, backbone.backbone.stages.3.3.layers.1.bn.running_var, backbone.backbone.stages.3.3.layers.2.conv.weight, backbone.backbone.stages.3.3.layers.2.bn.weight, backbone.backbone.stages.3.3.layers.2.bn.bias, backbone.backbone.stages.3.3.layers.2.bn.running_mean, backbone.backbone.stages.3.3.layers.2.bn.running_var, backbone.backbone.stages.3.3.layers.3.conv.weight, backbone.backbone.stages.3.3.layers.3.bn.weight, backbone.backbone.stages.3.3.layers.3.bn.bias, backbone.backbone.stages.3.3.layers.3.bn.running_mean, backbone.backbone.stages.3.3.layers.3.bn.running_var, backbone.backbone.stages.3.4.layers.0.conv.weight, backbone.backbone.stages.3.4.layers.0.bn.weight, backbone.backbone.stages.3.4.layers.0.bn.bias, backbone.backbone.stages.3.4.layers.0.bn.running_mean, backbone.backbone.stages.3.4.layers.0.bn.running_var, backbone.backbone.stages.3.4.layers.1.conv.weight, backbone.backbone.stages.3.4.layers.1.bn.weight, backbone.backbone.stages.3.4.layers.1.bn.bias, backbone.backbone.stages.3.4.layers.1.bn.running_mean, backbone.backbone.stages.3.4.layers.1.bn.running_var, backbone.backbone.stages.3.4.layers.2.conv.weight, backbone.backbone.stages.3.4.layers.2.bn.weight, backbone.backbone.stages.3.4.layers.2.bn.bias, backbone.backbone.stages.3.4.layers.2.bn.running_mean, backbone.backbone.stages.3.4.layers.2.bn.running_var, backbone.backbone.stages.3.4.layers.3.conv.weight, backbone.backbone.stages.3.4.layers.3.bn.weight, backbone.backbone.stages.3.4.layers.3.bn.bias, backbone.backbone.stages.3.4.layers.3.bn.running_mean, backbone.backbone.stages.3.4.layers.3.bn.running_var, backbone.backbone.stages.4.0.layers.0.conv.weight, backbone.backbone.stages.4.0.layers.0.bn.weight, backbone.backbone.stages.4.0.layers.0.bn.bias, backbone.backbone.stages.4.0.layers.0.bn.running_mean, backbone.backbone.stages.4.0.layers.0.bn.running_var, backbone.backbone.stages.4.0.layers.1.conv.weight, backbone.backbone.stages.4.0.layers.1.bn.weight, backbone.backbone.stages.4.0.layers.1.bn.bias, backbone.backbone.stages.4.0.layers.1.bn.running_mean, backbone.backbone.stages.4.0.layers.1.bn.running_var, backbone.backbone.stages.4.0.layers.2.conv.weight, backbone.backbone.stages.4.0.layers.2.bn.weight, backbone.backbone.stages.4.0.layers.2.bn.bias, backbone.backbone.stages.4.0.layers.2.bn.running_mean, backbone.backbone.stages.4.0.layers.2.bn.running_var, backbone.backbone.stages.4.0.layers.3.conv.weight, backbone.backbone.stages.4.0.layers.3.bn.weight, backbone.backbone.stages.4.0.layers.3.bn.bias, backbone.backbone.stages.4.0.layers.3.bn.running_mean, backbone.backbone.stages.4.0.layers.3.bn.running_var, backbone.backbone.stages.4.0.downsample.conv.weight, backbone.backbone.stages.4.0.downsample.bn.weight, backbone.backbone.stages.4.0.downsample.bn.bias, backbone.backbone.stages.4.0.downsample.bn.running_mean, backbone.backbone.stages.4.0.downsample.bn.running_var, backbone.backbone.stages.4.1.layers.0.conv.weight, backbone.backbone.stages.4.1.layers.0.bn.weight, backbone.backbone.stages.4.1.layers.0.bn.bias, backbone.backbone.stages.4.1.layers.0.bn.running_mean, backbone.backbone.stages.4.1.layers.0.bn.running_var, backbone.backbone.stages.4.1.layers.1.conv.weight, backbone.backbone.stages.4.1.layers.1.bn.weight, backbone.backbone.stages.4.1.layers.1.bn.bias, backbone.backbone.stages.4.1.layers.1.bn.running_mean, backbone.backbone.stages.4.1.layers.1.bn.running_var, backbone.backbone.stages.4.1.layers.2.conv.weight, backbone.backbone.stages.4.1.layers.2.bn.weight, backbone.backbone.stages.4.1.layers.2.bn.bias, backbone.backbone.stages.4.1.layers.2.bn.running_mean, backbone.backbone.stages.4.1.layers.2.bn.running_var, backbone.backbone.stages.4.1.layers.3.conv.weight, backbone.backbone.stages.4.1.layers.3.bn.weight, backbone.backbone.stages.4.1.layers.3.bn.bias, backbone.backbone.stages.4.1.layers.3.bn.running_mean, backbone.backbone.stages.4.1.layers.3.bn.running_var, backbone.backbone.stages.4.2.layers.0.conv.weight, backbone.backbone.stages.4.2.layers.0.bn.weight, backbone.backbone.stages.4.2.layers.0.bn.bias, backbone.backbone.stages.4.2.layers.0.bn.running_mean, backbone.backbone.stages.4.2.layers.0.bn.running_var, backbone.backbone.stages.4.2.layers.1.conv.weight, backbone.backbone.stages.4.2.layers.1.bn.weight, backbone.backbone.stages.4.2.layers.1.bn.bias, backbone.backbone.stages.4.2.layers.1.bn.running_mean, backbone.backbone.stages.4.2.layers.1.bn.running_var, backbone.backbone.stages.4.2.layers.2.conv.weight, backbone.backbone.stages.4.2.layers.2.bn.weight, backbone.backbone.stages.4.2.layers.2.bn.bias, backbone.backbone.stages.4.2.layers.2.bn.running_mean, backbone.backbone.stages.4.2.layers.2.bn.running_var, backbone.backbone.stages.4.2.layers.3.conv.weight, backbone.backbone.stages.4.2.layers.3.bn.weight, backbone.backbone.stages.4.2.layers.3.bn.bias, backbone.backbone.stages.4.2.layers.3.bn.running_mean, backbone.backbone.stages.4.2.layers.3.bn.running_var, backbone.arms.0.conv_layer.conv.weight, backbone.arms.0.conv_layer.bn.weight, backbone.arms.0.conv_layer.bn.bias, backbone.arms.0.conv_layer.bn.running_mean, backbone.arms.0.conv_layer.bn.running_var, backbone.arms.0.atten_conv_layer.1.conv.weight, backbone.arms.0.atten_conv_layer.1.bn.weight, backbone.arms.0.atten_conv_layer.1.bn.bias, backbone.arms.0.atten_conv_layer.1.bn.running_mean, backbone.arms.0.atten_conv_layer.1.bn.running_var, backbone.arms.1.conv_layer.conv.weight, backbone.arms.1.conv_layer.bn.weight, backbone.arms.1.conv_layer.bn.bias, backbone.arms.1.conv_layer.bn.running_mean, backbone.arms.1.conv_layer.bn.running_var, backbone.arms.1.atten_conv_layer.1.conv.weight, backbone.arms.1.atten_conv_layer.1.bn.weight, backbone.arms.1.atten_conv_layer.1.bn.bias, backbone.arms.1.atten_conv_layer.1.bn.running_mean, backbone.arms.1.atten_conv_layer.1.bn.running_var, backbone.convs.0.conv.weight, backbone.convs.0.bn.weight, backbone.convs.0.bn.bias, backbone.convs.0.bn.running_mean, backbone.convs.0.bn.running_var, backbone.convs.1.conv.weight, backbone.convs.1.bn.weight, backbone.convs.1.bn.bias, backbone.convs.1.bn.running_mean, backbone.convs.1.bn.running_var, backbone.conv_avg.conv.weight, backbone.conv_avg.bn.weight, backbone.conv_avg.bn.bias, backbone.conv_avg.bn.running_mean, backbone.conv_avg.bn.running_var, backbone.ffm.conv0.conv.weight, backbone.ffm.conv0.bn.weight, backbone.ffm.conv0.bn.bias, backbone.ffm.conv0.bn.running_mean, backbone.ffm.conv0.bn.running_var, backbone.ffm.attention.1.conv.weight, backbone.ffm.attention.2.conv.weight, decode_head.conv_seg.weight, decode_head.conv_seg.bias, decode_head.convs.0.conv.weight, decode_head.convs.0.bn.weight, decode_head.convs.0.bn.bias, decode_head.convs.0.bn.running_mean, decode_head.convs.0.bn.running_var, auxiliary_head.0.conv_seg.weight, auxiliary_head.0.conv_seg.bias, auxiliary_head.0.convs.0.conv.weight, auxiliary_head.0.convs.0.bn.weight, auxiliary_head.0.convs.0.bn.bias, auxiliary_head.0.convs.0.bn.running_mean, auxiliary_head.0.convs.0.bn.running_var, auxiliary_head.1.conv_seg.weight, auxiliary_head.1.conv_seg.bias, auxiliary_head.1.convs.0.conv.weight, auxiliary_head.1.convs.0.bn.weight, auxiliary_head.1.convs.0.bn.bias, auxiliary_head.1.convs.0.bn.running_mean, auxiliary_head.1.convs.0.bn.running_var, auxiliary_head.2.fusion_kernel, auxiliary_head.2.laplacian_kernel, auxiliary_head.2.conv_seg.weight, auxiliary_head.2.conv_seg.bias, auxiliary_head.2.convs.0.conv.weight, auxiliary_head.2.convs.0.bn.weight, auxiliary_head.2.convs.0.bn.bias, auxiliary_head.2.convs.0.bn.running_mean, auxiliary_head.2.convs.0.bn.running_var

[MengzhangLI]

Could you try to follow this link:

#1418 (comment)