RuntimeError: Error(s) in loading state_dict for DeepLab
rsuwa opened this issue · comments
When I ran python test.py -c config/config.yaml , I got the following error.
RuntimeError: Error(s) in loading state_dict for DeepLab:
Missing key(s) in state_dict: "backbone.layer0.0.weight", "backbone.layer0.1.weight", "backbone.layer0.1.bias", "backbone.layer0.1.running_mean", "backbone.layer0.1.running_var", "backbone.layer1.0.weight", "backbone.layer1.1.weight", "backbone.layer1.1.bias", "backbone.layer1.1.running_mean", "backbone.layer1.1.running_var", "backbone.layer2.0.weight", "backbone.layer2.1.weight", "backbone.layer2.1.bias", "backbone.layer2.1.running_mean", "backbone.layer2.1.running_var", "backbone.layer3.0.conv1.weight", "backbone.layer3.0.bn1.weight", "backbone.layer3.0.bn1.bias", "backbone.layer3.0.bn1.running_mean", "backbone.layer3.0.bn1.running_var", "backbone.layer3.0.conv2.weight", "backbone.layer3.0.bn2.weight", "backbone.layer3.0.bn2.bias", "backbone.layer3.0.bn2.running_mean", "backbone.layer3.0.bn2.running_var", "backbone.layer3.0.conv3.weight", "backbone.layer3.0.bn3.weight", "backbone.layer3.0.bn3.bias", "backbone.layer3.0.bn3.running_mean", "backbone.layer3.0.bn3.running_var", "backbone.layer3.0.downsample.0.weight", "backbone.layer3.0.downsample.1.weight", "backbone.layer3.0.downsample.1.bias", "backbone.layer3.0.downsample.1.running_mean", "backbone.layer3.0.downsample.1.running_var", "backbone.layer3.1.conv1.weight", "backbone.layer3.1.bn1.weight", "backbone.layer3.1.bn1.bias", "backbone.layer3.1.bn1.running_mean", "backbone.layer3.1.bn1.running_var", "backbone.layer3.1.conv2.weight", "backbone.layer3.1.bn2.weight", "backbone.layer3.1.bn2.bias", "backbone.layer3.1.bn2.running_mean", "backbone.layer3.1.bn2.running_var", "backbone.layer3.1.conv3.weight", "backbone.layer3.1.bn3.weight", "backbone.layer3.1.bn3.bias", "backbone.layer3.1.bn3.running_mean", "backbone.layer3.1.bn3.running_var", "backbone.layer3.2.conv1.weight", "backbone.layer3.2.bn1.weight", "backbone.layer3.2.bn1.bias", "backbone.layer3.2.bn1.running_mean", "backbone.layer3.2.bn1.running_var", "backbone.layer3.2.conv2.weight", "backbone.layer3.2.bn2.weight", "backbone.layer3.2.bn2.bias", "backbone.layer3.2.bn2.running_mean", "backbone.layer3.2.bn2.running_var", "backbone.layer3.2.conv3.weight", "backbone.layer3.2.bn3.weight", "backbone.layer3.2.bn3.bias", "backbone.layer3.2.bn3.running_mean", "backbone.layer3.2.bn3.running_var", "backbone.layer4.0.conv1.weight", "backbone.layer4.0.bn1.weight", "backbone.layer4.0.bn1.bias", "backbone.layer4.0.bn1.running_mean", "backbone.layer4.0.bn1.running_var", "backbone.layer4.0.conv2.weight", "backbone.layer4.0.bn2.weight", "backbone.layer4.0.bn2.bias", "backbone.layer4.0.bn2.running_mean", "backbone.layer4.0.bn2.running_var", "backbone.layer4.0.conv3.weight", "backbone.layer4.0.bn3.weight", "backbone.layer4.0.bn3.bias", "backbone.layer4.0.bn3.running_mean", "backbone.layer4.0.bn3.running_var", "backbone.layer4.0.downsample.0.weight", "backbone.layer4.0.downsample.1.weight", "backbone.layer4.0.downsample.1.bias", "backbone.layer4.0.downsample.1.running_mean", "backbone.layer4.0.downsample.1.running_var", "backbone.layer4.1.conv1.weight", "backbone.layer4.1.bn1.weight", "backbone.layer4.1.bn1.bias", "backbone.layer4.1.bn1.running_mean", "backbone.layer4.1.bn1.running_var", "backbone.layer4.1.conv2.weight", "backbone.layer4.1.bn2.weight", "backbone.layer4.1.bn2.bias", "backbone.layer4.1.bn2.running_mean", "backbone.layer4.1.bn2.running_var", "backbone.layer4.1.conv3.weight", "backbone.layer4.1.bn3.weight", "backbone.layer4.1.bn3.bias", "backbone.layer4.1.bn3.running_mean", "backbone.layer4.1.bn3.running_var", "backbone.layer4.2.conv1.weight", "backbone.layer4.2.bn1.weight", "backbone.layer4.2.bn1.bias", "backbone.layer4.2.bn1.running_mean", "backbone.layer4.2.bn1.running_var", "backbone.layer4.2.conv2.weight", "backbone.layer4.2.bn2.weight", "backbone.layer4.2.bn2.bias", "backbone.layer4.2.bn2.running_mean", "backbone.layer4.2.bn2.running_var", "backbone.layer4.2.conv3.weight", "backbone.layer4.2.bn3.weight", "backbone.layer4.2.bn3.bias", "backbone.layer4.2.bn3.running_mean", "backbone.layer4.2.bn3.running_var", "backbone.layer4.3.conv1.weight", "backbone.layer4.3.bn1.weight", "backbone.layer4.3.bn1.bias", "backbone.layer4.3.bn1.running_mean", "backbone.layer4.3.bn1.running_var", "backbone.layer4.3.conv2.weight", "backbone.layer4.3.bn2.weight", "backbone.layer4.3.bn2.bias", "backbone.layer4.3.bn2.running_mean", "backbone.layer4.3.bn2.running_var", "backbone.layer4.3.conv3.weight", "backbone.layer4.3.bn3.weight", "backbone.layer4.3.bn3.bias", "backbone.layer4.3.bn3.running_mean", "backbone.layer4.3.bn3.running_var", "backbone.layer5.0.conv1.weight", "backbone.layer5.0.bn1.weight", "backbone.layer5.0.bn1.bias", "backbone.layer5.0.bn1.running_mean", "backbone.layer5.0.bn1.running_var", "backbone.layer5.0.conv2.weight", "backbone.layer5.0.bn2.weight", "backbone.layer5.0.bn2.bias", "backbone.layer5.0.bn2.running_mean", "backbone.layer5.0.bn2.running_var", "backbone.layer5.0.conv3.weight", "backbone.layer5.0.bn3.weight", "backbone.layer5.0.bn3.bias", "backbone.layer5.0.bn3.running_mean", "backbone.layer5.0.bn3.running_var", "backbone.layer5.0.downsample.0.weight", "backbone.layer5.0.downsample.1.weight", "backbone.layer5.0.downsample.1.bias", "backbone.layer5.0.downsample.1.running_mean", "backbone.layer5.0.downsample.1.running_var", "backbone.layer5.1.conv1.weight", "backbone.layer5.1.bn1.weight", "backbone.layer5.1.bn1.bias", "backbone.layer5.1.bn1.running_mean", "backbone.layer5.1.bn1.running_var", "backbone.layer5.1.conv2.weight", "backbone.layer5.1.bn2.weight", "backbone.layer5.1.bn2.bias", "backbone.layer5.1.bn2.running_mean", "backbone.layer5.1.bn2.running_var", "backbone.layer5.1.conv3.weight", "backbone.layer5.1.bn3.weight", "backbone.layer5.1.bn3.bias", "backbone.layer5.1.bn3.running_mean", "backbone.layer5.1.bn3.running_var", "backbone.layer5.2.conv1.weight", "backbone.layer5.2.bn1.weight", "backbone.layer5.2.bn1.bias", "backbone.layer5.2.bn1.running_mean", "backbone.layer5.2.bn1.running_var", "backbone.layer5.2.conv2.weight", "backbone.layer5.2.bn2.weight", "backbone.layer5.2.bn2.bias", "backbone.layer5.2.bn2.running_mean", "backbone.layer5.2.bn2.running_var", "backbone.layer5.2.conv3.weight", "backbone.layer5.2.bn3.weight", "backbone.layer5.2.bn3.bias", "backbone.layer5.2.bn3.running_mean", "backbone.layer5.2.bn3.running_var", "backbone.layer5.3.conv1.weight", "backbone.layer5.3.bn1.weight", "backbone.layer5.3.bn1.bias", "backbone.layer5.3.bn1.running_mean", "backbone.layer5.3.bn1.running_var", "backbone.layer5.3.conv2.weight", "backbone.layer5.3.bn2.weight", "backbone.layer5.3.bn2.bias", "backbone.layer5.3.bn2.running_mean", "backbone.layer5.3.bn2.running_var", "backbone.layer5.3.conv3.weight", "backbone.layer5.3.bn3.weight", "backbone.layer5.3.bn3.bias", "backbone.layer5.3.bn3.running_mean", "backbone.layer5.3.bn3.running_var", "backbone.layer5.4.conv1.weight", "backbone.layer5.4.bn1.weight", "backbone.layer5.4.bn1.bias", "backbone.layer5.4.bn1.running_mean", "backbone.layer5.4.bn1.running_var", "backbone.layer5.4.conv2.weight", "backbone.layer5.4.bn2.weight", "backbone.layer5.4.bn2.bias", "backbone.layer5.4.bn2.running_mean", "backbone.layer5.4.bn2.running_var", "backbone.layer5.4.conv3.weight", "backbone.layer5.4.bn3.weight", "backbone.layer5.4.bn3.bias", "backbone.layer5.4.bn3.running_mean", "backbone.layer5.4.bn3.running_var", "backbone.layer5.5.conv1.weight", "backbone.layer5.5.bn1.weight", "backbone.layer5.5.bn1.bias", "backbone.layer5.5.bn1.running_mean", "backbone.layer5.5.bn1.running_var", "backbone.layer5.5.conv2.weight", "backbone.layer5.5.bn2.weight", "backbone.layer5.5.bn2.bias", "backbone.layer5.5.bn2.running_mean", "backbone.layer5.5.bn2.running_var", "backbone.layer5.5.conv3.weight", "backbone.layer5.5.bn3.weight", "backbone.layer5.5.bn3.bias", "backbone.layer5.5.bn3.running_mean", "backbone.layer5.5.bn3.running_var", "backbone.layer6.0.conv1.weight", "backbone.layer6.0.bn1.weight", "backbone.layer6.0.bn1.bias", "backbone.layer6.0.bn1.running_mean", "backbone.layer6.0.bn1.running_var", "backbone.layer6.0.conv2.weight", "backbone.layer6.0.bn2.weight", "backbone.layer6.0.bn2.bias", "backbone.layer6.0.bn2.running_mean", "backbone.layer6.0.bn2.running_var", "backbone.layer6.0.conv3.weight", "backbone.layer6.0.bn3.weight", "backbone.layer6.0.bn3.bias", "backbone.layer6.0.bn3.running_mean", "backbone.layer6.0.bn3.running_var", "backbone.layer6.0.downsample.0.weight", "backbone.layer6.0.downsample.1.weight", "backbone.layer6.0.downsample.1.bias", "backbone.layer6.0.downsample.1.running_mean", "backbone.layer6.0.downsample.1.running_var", "backbone.layer6.1.conv1.weight", "backbone.layer6.1.bn1.weight", "backbone.layer6.1.bn1.bias", "backbone.layer6.1.bn1.running_mean", "backbone.layer6.1.bn1.running_var", "backbone.layer6.1.conv2.weight", "backbone.layer6.1.bn2.weight", "backbone.layer6.1.bn2.bias", "backbone.layer6.1.bn2.running_mean", "backbone.layer6.1.bn2.running_var", "backbone.layer6.1.conv3.weight", "backbone.layer6.1.bn3.weight", "backbone.layer6.1.bn3.bias", "backbone.layer6.1.bn3.running_mean", "backbone.layer6.1.bn3.running_var", "backbone.layer6.2.conv1.weight", "backbone.layer6.2.bn1.weight", "backbone.layer6.2.bn1.bias", "backbone.layer6.2.bn1.running_mean", "backbone.layer6.2.bn1.running_var", "backbone.layer6.2.conv2.weight", "backbone.layer6.2.bn2.weight", "backbone.layer6.2.bn2.bias", "backbone.layer6.2.bn2.running_mean", "backbone.layer6.2.bn2.running_var", "backbone.layer6.2.conv3.weight", "backbone.layer6.2.bn3.weight", "backbone.layer6.2.bn3.bias", "backbone.layer6.2.bn3.running_mean", "backbone.layer6.2.bn3.running_var", "backbone.layer7.0.weight", "backbone.layer7.1.weight", "backbone.layer7.1.bias", "backbone.layer7.1.running_mean", "backbone.layer7.1.running_var", "backbone.layer8.0.weight", "backbone.layer8.1.weight", "backbone.layer8.1.bias", "backbone.layer8.1.running_mean", "backbone.layer8.1.running_var", "aspp.aspp1.atrous_conv.weight", "aspp.aspp1.bn.weight", "aspp.aspp1.bn.bias", "aspp.aspp1.bn.running_mean", "aspp.aspp1.bn.running_var", "aspp.aspp2.atrous_conv.weight", "aspp.aspp2.bn.weight", "aspp.aspp2.bn.bias", "aspp.aspp2.bn.running_mean", "aspp.aspp2.bn.running_var", "aspp.aspp3.atrous_conv.weight", "aspp.aspp3.bn.weight", "aspp.aspp3.bn.bias", "aspp.aspp3.bn.running_mean", "aspp.aspp3.bn.running_var", "aspp.aspp4.atrous_conv.weight", "aspp.aspp4.bn.weight", "aspp.aspp4.bn.bias", "aspp.aspp4.bn.running_mean", "aspp.aspp4.bn.running_var", "aspp.global_avg_pool.1.weight", "aspp.global_avg_pool.2.weight", "aspp.global_avg_pool.2.bias", "aspp.global_avg_pool.2.running_mean", "aspp.global_avg_pool.2.running_var", "aspp.conv1.weight", "aspp.bn1.weight", "aspp.bn1.bias", "aspp.bn1.running_mean", "aspp.bn1.running_var", "decoder.conv1.weight", "decoder.bn1.weight", "decoder.bn1.bias", "decoder.bn1.running_mean", "decoder.bn1.running_var", "decoder.last_conv.0.weight", "decoder.last_conv.1.weight", "decoder.last_conv.1.bias", "decoder.last_conv.1.running_mean", "decoder.last_conv.1.running_var", "decoder.last_conv.4.weight", "decoder.last_conv.5.weight", "decoder.last_conv.5.bias", "decoder.last_conv.5.running_mean", "decoder.last_conv.5.running_var", "decoder.last_conv.7.weight", "decoder.last_conv.7.bias".
Unexpected key(s) in state_dict: "module.backbone.layer0.0.weight", "module.backbone.layer0.1.weight", "module.backbone.layer0.1.bias", "module.backbone.layer0.1.running_mean", "module.backbone.layer0.1.running_var", "module.backbone.layer0.1.num_batches_tracked", "module.backbone.layer1.0.weight", "module.backbone.layer1.1.weight", "module.backbone.layer1.1.bias", "module.backbone.layer1.1.running_mean", "module.backbone.layer1.1.running_var", "module.backbone.layer1.1.num_batches_tracked", "module.backbone.layer2.0.weight", "module.backbone.layer2.1.weight", "module.backbone.layer2.1.bias", "module.backbone.layer2.1.running_mean", "module.backbone.layer2.1.running_var", "module.backbone.layer2.1.num_batches_tracked", "module.backbone.layer3.0.conv1.weight", "module.backbone.layer3.0.bn1.weight", "module.backbone.layer3.0.bn1.bias", "module.backbone.layer3.0.bn1.running_mean", "module.backbone.layer3.0.bn1.running_var", "module.backbone.layer3.0.bn1.num_batches_tracked", "module.backbone.layer3.0.conv2.weight", "module.backbone.layer3.0.bn2.weight", "module.backbone.layer3.0.bn2.bias", "module.backbone.layer3.0.bn2.running_mean", "module.backbone.layer3.0.bn2.running_var", "module.backbone.layer3.0.bn2.num_batches_tracked", "module.backbone.layer3.0.conv3.weight", "module.backbone.layer3.0.bn3.weight", "module.backbone.layer3.0.bn3.bias", "module.backbone.layer3.0.bn3.running_mean", "module.backbone.layer3.0.bn3.running_var", "module.backbone.layer3.0.bn3.num_batches_tracked", "module.backbone.layer3.0.downsample.0.weight", "module.backbone.layer3.0.downsample.1.weight", "module.backbone.layer3.0.downsample.1.bias", "module.backbone.layer3.0.downsample.1.running_mean", "module.backbone.layer3.0.downsample.1.running_var", "module.backbone.layer3.0.downsample.1.num_batches_tracked", "module.backbone.layer3.1.conv1.weight", "module.backbone.layer3.1.bn1.weight", "module.backbone.layer3.1.bn1.bias", "module.backbone.layer3.1.bn1.running_mean", "module.backbone.layer3.1.bn1.running_var", "module.backbone.layer3.1.bn1.num_batches_tracked", "module.backbone.layer3.1.conv2.weight", "module.backbone.layer3.1.bn2.weight", "module.backbone.layer3.1.bn2.bias", "module.backbone.layer3.1.bn2.running_mean", "module.backbone.layer3.1.bn2.running_var", "module.backbone.layer3.1.bn2.num_batches_tracked", "module.backbone.layer3.1.conv3.weight", "module.backbone.layer3.1.bn3.weight", "module.backbone.layer3.1.bn3.bias", "module.backbone.layer3.1.bn3.running_mean", "module.backbone.layer3.1.bn3.running_var", "module.backbone.layer3.1.bn3.num_batches_tracked", "module.backbone.layer3.2.conv1.weight", "module.backbone.layer3.2.bn1.weight", "module.backbone.layer3.2.bn1.bias", "module.backbone.layer3.2.bn1.running_mean", "module.backbone.layer3.2.bn1.running_var", "module.backbone.layer3.2.bn1.num_batches_tracked", "module.backbone.layer3.2.conv2.weight", "module.backbone.layer3.2.bn2.weight", "module.backbone.layer3.2.bn2.bias", "module.backbone.layer3.2.bn2.running_mean", "module.backbone.layer3.2.bn2.running_var", "module.backbone.layer3.2.bn2.num_batches_tracked", "module.backbone.layer3.2.conv3.weight", "module.backbone.layer3.2.bn3.weight", "module.backbone.layer3.2.bn3.bias", "module.backbone.layer3.2.bn3.running_mean", "module.backbone.layer3.2.bn3.running_var", "module.backbone.layer3.2.bn3.num_batches_tracked", "module.backbone.layer4.0.conv1.weight", "module.backbone.layer4.0.bn1.weight", "module.backbone.layer4.0.bn1.bias", "module.backbone.layer4.0.bn1.running_mean", "module.backbone.layer4.0.bn1.running_var", "module.backbone.layer4.0.bn1.num_batches_tracked", "module.backbone.layer4.0.conv2.weight", "module.backbone.layer4.0.bn2.weight", "module.backbone.layer4.0.bn2.bias", "module.backbone.layer4.0.bn2.running_mean", "module.backbone.layer4.0.bn2.running_var", "module.backbone.layer4.0.bn2.num_batches_tracked", "module.backbone.layer4.0.conv3.weight", "module.backbone.layer4.0.bn3.weight", "module.backbone.layer4.0.bn3.bias", "module.backbone.layer4.0.bn3.running_mean", "module.backbone.layer4.0.bn3.running_var", "module.backbone.layer4.0.bn3.num_batches_tracked", "module.backbone.layer4.0.downsample.0.weight", "module.backbone.layer4.0.downsample.1.weight", "module.backbone.layer4.0.downsample.1.bias", "module.backbone.layer4.0.downsample.1.running_mean", "module.backbone.layer4.0.downsample.1.running_var", "module.backbone.layer4.0.downsample.1.num_batches_tracked", "module.backbone.layer4.1.conv1.weight", "module.backbone.layer4.1.bn1.weight", "module.backbone.layer4.1.bn1.bias", "module.backbone.layer4.1.bn1.running_mean", "module.backbone.layer4.1.bn1.running_var", "module.backbone.layer4.1.bn1.num_batches_tracked", "module.backbone.layer4.1.conv2.weight", "module.backbone.layer4.1.bn2.weight", "module.backbone.layer4.1.bn2.bias", "module.backbone.layer4.1.bn2.running_mean", "module.backbone.layer4.1.bn2.running_var", "module.backbone.layer4.1.bn2.num_batches_tracked", "module.backbone.layer4.1.conv3.weight", "module.backbone.layer4.1.bn3.weight", "module.backbone.layer4.1.bn3.bias", "module.backbone.layer4.1.bn3.running_mean", "module.backbone.layer4.1.bn3.running_var", "module.backbone.layer4.1.bn3.num_batches_tracked", "module.backbone.layer4.2.conv1.weight", "module.backbone.layer4.2.bn1.weight", "module.backbone.layer4.2.bn1.bias", "module.backbone.layer4.2.bn1.running_mean", "module.backbone.layer4.2.bn1.running_var", "module.backbone.layer4.2.bn1.num_batches_tracked", "module.backbone.layer4.2.conv2.weight", "module.backbone.layer4.2.bn2.weight", "module.backbone.layer4.2.bn2.bias", "module.backbone.layer4.2.bn2.running_mean", "module.backbone.layer4.2.bn2.running_var", "module.backbone.layer4.2.bn2.num_batches_tracked", "module.backbone.layer4.2.conv3.weight", "module.backbone.layer4.2.bn3.weight", "module.backbone.layer4.2.bn3.bias", "module.backbone.layer4.2.bn3.running_mean", "module.backbone.layer4.2.bn3.running_var", "module.backbone.layer4.2.bn3.num_batches_tracked", "module.backbone.layer4.3.conv1.weight", "module.backbone.layer4.3.bn1.weight", "module.backbone.layer4.3.bn1.bias", "module.backbone.layer4.3.bn1.running_mean", "module.backbone.layer4.3.bn1.running_var", "module.backbone.layer4.3.bn1.num_batches_tracked", "module.backbone.layer4.3.conv2.weight", "module.backbone.layer4.3.bn2.weight", "module.backbone.layer4.3.bn2.bias", "module.backbone.layer4.3.bn2.running_mean", "module.backbone.layer4.3.bn2.running_var", "module.backbone.layer4.3.bn2.num_batches_tracked", "module.backbone.layer4.3.conv3.weight", "module.backbone.layer4.3.bn3.weight", "module.backbone.layer4.3.bn3.bias", "module.backbone.layer4.3.bn3.running_mean", "module.backbone.layer4.3.bn3.running_var", "module.backbone.layer4.3.bn3.num_batches_tracked", "module.backbone.layer5.0.conv1.weight", "module.backbone.layer5.0.bn1.weight", "module.backbone.layer5.0.bn1.bias", "module.backbone.layer5.0.bn1.running_mean", "module.backbone.layer5.0.bn1.running_var", "module.backbone.layer5.0.bn1.num_batches_tracked", "module.backbone.layer5.0.conv2.weight", "module.backbone.layer5.0.bn2.weight", "module.backbone.layer5.0.bn2.bias", "module.backbone.layer5.0.bn2.running_mean", "module.backbone.layer5.0.bn2.running_var", "module.backbone.layer5.0.bn2.num_batches_tracked", "module.backbone.layer5.0.conv3.weight", "module.backbone.layer5.0.bn3.weight", "module.backbone.layer5.0.bn3.bias", "module.backbone.layer5.0.bn3.running_mean", "module.backbone.layer5.0.bn3.running_var", "module.backbone.layer5.0.bn3.num_batches_tracked", "module.backbone.layer5.0.downsample.0.weight", "module.backbone.layer5.0.downsample.1.weight", "module.backbone.layer5.0.downsample.1.bias", "module.backbone.layer5.0.downsample.1.running_mean", "module.backbone.layer5.0.downsample.1.running_var", "module.backbone.layer5.0.downsample.1.num_batches_tracked", "module.backbone.layer5.1.conv1.weight", "module.backbone.layer5.1.bn1.weight", "module.backbone.layer5.1.bn1.bias", "module.backbone.layer5.1.bn1.running_mean", "module.backbone.layer5.1.bn1.running_var", "module.backbone.layer5.1.bn1.num_batches_tracked", "module.backbone.layer5.1.conv2.weight", "module.backbone.layer5.1.bn2.weight", "module.backbone.layer5.1.bn2.bias", "module.backbone.layer5.1.bn2.running_mean", "module.backbone.layer5.1.bn2.running_var", "module.backbone.layer5.1.bn2.num_batches_tracked", "module.backbone.layer5.1.conv3.weight", "module.backbone.layer5.1.bn3.weight", "module.backbone.layer5.1.bn3.bias", "module.backbone.layer5.1.bn3.running_mean", "module.backbone.layer5.1.bn3.running_var", "module.backbone.layer5.1.bn3.num_batches_tracked", "module.backbone.layer5.2.conv1.weight", "module.backbone.layer5.2.bn1.weight", "module.backbone.layer5.2.bn1.bias", "module.backbone.layer5.2.bn1.running_mean", "module.backbone.layer5.2.bn1.running_var", "module.backbone.layer5.2.bn1.num_batches_tracked", "module.backbone.layer5.2.conv2.weight", "module.backbone.layer5.2.bn2.weight", "module.backbone.layer5.2.bn2.bias", "module.backbone.layer5.2.bn2.running_mean", "module.backbone.layer5.2.bn2.running_var", "module.backbone.layer5.2.bn2.num_batches_tracked", "module.backbone.layer5.2.conv3.weight", "module.backbone.layer5.2.bn3.weight", "module.backbone.layer5.2.bn3.bias", "module.backbone.layer5.2.bn3.running_mean", "module.backbone.layer5.2.bn3.running_var", "module.backbone.layer5.2.bn3.num_batches_tracked", "module.backbone.layer5.3.conv1.weight", "module.backbone.layer5.3.bn1.weight", "module.backbone.layer5.3.bn1.bias", "module.backbone.layer5.3.bn1.running_mean", "module.backbone.layer5.3.bn1.running_var", "module.backbone.layer5.3.bn1.num_batches_tracked", "module.backbone.layer5.3.conv2.weight", "module.backbone.layer5.3.bn2.weight", "module.backbone.layer5.3.bn2.bias", "module.backbone.layer5.3.bn2.running_mean", "module.backbone.layer5.3.bn2.running_var", "module.backbone.layer5.3.bn2.num_batches_tracked", "module.backbone.layer5.3.conv3.weight", "module.backbone.layer5.3.bn3.weight", "module.backbone.layer5.3.bn3.bias", "module.backbone.layer5.3.bn3.running_mean", "module.backbone.layer5.3.bn3.running_var", "module.backbone.layer5.3.bn3.num_batches_tracked", "module.backbone.layer5.4.conv1.weight", "module.backbone.layer5.4.bn1.weight", "module.backbone.layer5.4.bn1.bias", "module.backbone.layer5.4.bn1.running_mean", "module.backbone.layer5.4.bn1.running_var", "module.backbone.layer5.4.bn1.num_batches_tracked", "module.backbone.layer5.4.conv2.weight", "module.backbone.layer5.4.bn2.weight", "module.backbone.layer5.4.bn2.bias", "module.backbone.layer5.4.bn2.running_mean", "module.backbone.layer5.4.bn2.running_var", "module.backbone.layer5.4.bn2.num_batches_tracked", "module.backbone.layer5.4.conv3.weight", "module.backbone.layer5.4.bn3.weight", "module.backbone.layer5.4.bn3.bias", "module.backbone.layer5.4.bn3.running_mean", "module.backbone.layer5.4.bn3.running_var", "module.backbone.layer5.4.bn3.num_batches_tracked", "module.backbone.layer5.5.conv1.weight", "module.backbone.layer5.5.bn1.weight", "module.backbone.layer5.5.bn1.bias", "module.backbone.layer5.5.bn1.running_mean", "module.backbone.layer5.5.bn1.running_var", "module.backbone.layer5.5.bn1.num_batches_tracked", "module.backbone.layer5.5.conv2.weight", "module.backbone.layer5.5.bn2.weight", "module.backbone.layer5.5.bn2.bias", "module.backbone.layer5.5.bn2.running_mean", "module.backbone.layer5.5.bn2.running_var", "module.backbone.layer5.5.bn2.num_batches_tracked", "module.backbone.layer5.5.conv3.weight", "module.backbone.layer5.5.bn3.weight", "module.backbone.layer5.5.bn3.bias", "module.backbone.layer5.5.bn3.running_mean", "module.backbone.layer5.5.bn3.running_var", "module.backbone.layer5.5.bn3.num_batches_tracked", "module.backbone.layer6.0.conv1.weight", "module.backbone.layer6.0.bn1.weight", "module.backbone.layer6.0.bn1.bias", "module.backbone.layer6.0.bn1.running_mean", "module.backbone.layer6.0.bn1.running_var", "module.backbone.layer6.0.bn1.num_batches_tracked", "module.backbone.layer6.0.conv2.weight", "module.backbone.layer6.0.bn2.weight", "module.backbone.layer6.0.bn2.bias", "module.backbone.layer6.0.bn2.running_mean", "module.backbone.layer6.0.bn2.running_var", "module.backbone.layer6.0.bn2.num_batches_tracked", "module.backbone.layer6.0.conv3.weight", "module.backbone.layer6.0.bn3.weight", "module.backbone.layer6.0.bn3.bias", "module.backbone.layer6.0.bn3.running_mean", "module.backbone.layer6.0.bn3.running_var", "module.backbone.layer6.0.bn3.num_batches_tracked", "module.backbone.layer6.0.downsample.0.weight", "module.backbone.layer6.0.downsample.1.weight", "module.backbone.layer6.0.downsample.1.bias", "module.backbone.layer6.0.downsample.1.running_mean", "module.backbone.layer6.0.downsample.1.running_var", "module.backbone.layer6.0.downsample.1.num_batches_tracked", "module.backbone.layer6.1.conv1.weight", "module.backbone.layer6.1.bn1.weight", "module.backbone.layer6.1.bn1.bias", "module.backbone.layer6.1.bn1.running_mean", "module.backbone.layer6.1.bn1.running_var", "module.backbone.layer6.1.bn1.num_batches_tracked", "module.backbone.layer6.1.conv2.weight", "module.backbone.layer6.1.bn2.weight", "module.backbone.layer6.1.bn2.bias", "module.backbone.layer6.1.bn2.running_mean", "module.backbone.layer6.1.bn2.running_var", "module.backbone.layer6.1.bn2.num_batches_tracked", "module.backbone.layer6.1.conv3.weight", "module.backbone.layer6.1.bn3.weight", "module.backbone.layer6.1.bn3.bias", "module.backbone.layer6.1.bn3.running_mean", "module.backbone.layer6.1.bn3.running_var", "module.backbone.layer6.1.bn3.num_batches_tracked", "module.backbone.layer6.2.conv1.weight", "module.backbone.layer6.2.bn1.weight", "module.backbone.layer6.2.bn1.bias", "module.backbone.layer6.2.bn1.running_mean", "module.backbone.layer6.2.bn1.running_var", "module.backbone.layer6.2.bn1.num_batches_tracked", "module.backbone.layer6.2.conv2.weight", "module.backbone.layer6.2.bn2.weight", "module.backbone.layer6.2.bn2.bias", "module.backbone.layer6.2.bn2.running_mean", "module.backbone.layer6.2.bn2.running_var", "module.backbone.layer6.2.bn2.num_batches_tracked", "module.backbone.layer6.2.conv3.weight", "module.backbone.layer6.2.bn3.weight", "module.backbone.layer6.2.bn3.bias", "module.backbone.layer6.2.bn3.running_mean", "module.backbone.layer6.2.bn3.running_var", "module.backbone.layer6.2.bn3.num_batches_tracked", "module.backbone.layer7.0.weight", "module.backbone.layer7.1.weight", "module.backbone.layer7.1.bias", "module.backbone.layer7.1.running_mean", "module.backbone.layer7.1.running_var", "module.backbone.layer7.1.num_batches_tracked", "module.backbone.layer8.0.weight", "module.backbone.layer8.1.weight", "module.backbone.layer8.1.bias", "module.backbone.layer8.1.running_mean", "module.backbone.layer8.1.running_var", "module.backbone.layer8.1.num_batches_tracked", "module.aspp.aspp1.atrous_conv.weight", "module.aspp.aspp1.bn.weight", "module.aspp.aspp1.bn.bias", "module.aspp.aspp1.bn.running_mean", "module.aspp.aspp1.bn.running_var", "module.aspp.aspp1.bn.num_batches_tracked", "module.aspp.aspp2.atrous_conv.weight", "module.aspp.aspp2.bn.weight", "module.aspp.aspp2.bn.bias", "module.aspp.aspp2.bn.running_mean", "module.aspp.aspp2.bn.running_var", "module.aspp.aspp2.bn.num_batches_tracked", "module.aspp.aspp3.atrous_conv.weight", "module.aspp.aspp3.bn.weight", "module.aspp.aspp3.bn.bias", "module.aspp.aspp3.bn.running_mean", "module.aspp.aspp3.bn.running_var", "module.aspp.aspp3.bn.num_batches_tracked", "module.aspp.aspp4.atrous_conv.weight", "module.aspp.aspp4.bn.weight", "module.aspp.aspp4.bn.bias", "module.aspp.aspp4.bn.running_mean", "module.aspp.aspp4.bn.running_var", "module.aspp.aspp4.bn.num_batches_tracked", "module.aspp.global_avg_pool.1.weight", "module.aspp.global_avg_pool.2.weight", "module.aspp.global_avg_pool.2.bias", "module.aspp.global_avg_pool.2.running_mean", "module.aspp.global_avg_pool.2.running_var", "module.aspp.global_avg_pool.2.num_batches_tracked", "module.aspp.conv1.weight", "module.aspp.bn1.weight", "module.aspp.bn1.bias", "module.aspp.bn1.running_mean", "module.aspp.bn1.running_var", "module.aspp.bn1.num_batches_tracked", "module.decoder.conv1.weight", "module.decoder.bn1.weight", "module.decoder.bn1.bias", "module.decoder.bn1.running_mean", "module.decoder.bn1.running_var", "module.decoder.bn1.num_batches_tracked", "module.decoder.last_conv.0.weight", "module.decoder.last_conv.1.weight", "module.decoder.last_conv.1.bias", "module.decoder.last_conv.1.running_mean", "module.decoder.last_conv.1.running_var", "module.decoder.last_conv.1.num_batches_tracked", "module.decoder.last_conv.4.weight", "module.decoder.last_conv.5.weight", "module.decoder.last_conv.5.bias", "module.decoder.last_conv.5.running_mean", "module.decoder.last_conv.5.running_var", "module.decoder.last_conv.5.num_batches_tracked", "module.decoder.last_conv.7.weight", "module.decoder.last_conv.7.bias".
Which weights were used? Could you paste the full error msg?
Since we were using dataparallel to train on multiple GPUs, we wrapped the model in dataparallel when saving/loading the checkpoints. Which is why the checkpoint contains the module. prefix in front of all the layers.
To solve the error, we need to wrap the model in dataparallel before loading the weights:
model = dataparallel(model)