search

jiangxiluning / FOTS.PyTorch

FOTS Pytorch Implementation
BSD 3-Clause "New" or "Revised" License
640 stars 193 forks source link

RuntimeError: Error(s) in loading state_dict for FOTSModel #43

Open donglin8506 opened 5 years ago

donglin8506 commented 5 years ago

RuntimeError: Error(s) in loading state_dict for FOTSModel: Missing key(s) in state_dict: "sharedConv.backbone.conv1.weight", "sharedConv.backbone.bn1.weight", "sharedConv.backbone.bn1.bias", "sharedConv.backbone.bn1.running_mean", "sharedConv.backbone.bn1.running_var", "sharedConv.backbone.layer1.0.conv1.weight", "sharedConv.backbone.layer1.0.bn1.weight", "sharedConv.backbone.layer1.0.bn1.bias", "sharedConv.backbone.layer1.0.bn1.running_mean", "sharedConv.backbone.layer1.0.bn1.running_var", "sharedConv.backbone.layer1.0.conv2.weight", "sharedConv.backbone.layer1.0.bn2.weight", "sharedConv.backbone.layer1.0.bn2.bias", "sharedConv.backbone.layer1.0.bn2.running_mean", "sharedConv.backbone.layer1.0.bn2.running_var", "sharedConv.backbone.layer1.0.conv3.weight", "sharedConv.backbone.layer1.0.bn3.weight", "sharedConv.backbone.layer1.0.bn3.bias", "sharedConv.backbone.layer1.0.bn3.running_mean", "sharedConv.backbone.layer1.0.bn3.running_var", "sharedConv.backbone.layer1.0.downsample.0.weight", "sharedConv.backbone.layer1.0.downsample.1.weight", "sharedConv.backbone.layer1.0.downsample.1.bias", "sharedConv.backbone.layer1.0.downsample.1.running_mean", "sharedConv.backbone.layer1.0.downsample.1.running_var", "sharedConv.backbone.layer1.1.conv1.weight", "sharedConv.backbone.layer1.1.bn1.weight", "sharedConv.backbone.layer1.1.bn1.bias", "sharedConv.backbone.layer1.1.bn1.running_mean", "sharedConv.backbone.layer1.1.bn1.running_var", "sharedConv.backbone.layer1.1.conv2.weight", "sharedConv.backbone.layer1.1.bn2.weight", "sharedConv.backbone.layer1.1.bn2.bias", "sharedConv.backbone.layer1.1.bn2.running_mean", "sharedConv.backbone.layer1.1.bn2.running_var", "sharedConv.backbone.layer1.1.conv3.weight", "sharedConv.backbone.layer1.1.bn3.weight", "sharedConv.backbone.layer1.1.bn3.bias", "sharedConv.backbone.layer1.1.bn3.running_mean", "sharedConv.backbone.layer1.1.bn3.running_var", "sharedConv.backbone.layer1.2.conv1.weight", "sharedConv.backbone.layer1.2.bn1.weight", "sharedConv.backbone.layer1.2.bn1.bias", "sharedConv.backbone.layer1.2.bn1.running_mean", "sharedConv.backbone.layer1.2.bn1.running_var", "sharedConv.backbone.layer1.2.conv2.weight", "sharedConv.backbone.layer1.2.bn2.weight", "sharedConv.backbone.layer1.2.bn2.bias", "sharedConv.backbone.layer1.2.bn2.running_mean", "sharedConv.backbone.layer1.2.bn2.running_var", "sharedConv.backbone.layer1.2.conv3.weight", "sharedConv.backbone.layer1.2.bn3.weight", "sharedConv.backbone.layer1.2.bn3.bias", "sharedConv.backbone.layer1.2.bn3.running_mean", "sharedConv.backbone.layer1.2.bn3.running_var", "sharedConv.backbone.layer2.0.conv1.weight", "sharedConv.backbone.layer2.0.bn1.weight", "sharedConv.backbone.layer2.0.bn1.bias", "sharedConv.backbone.layer2.0.bn1.running_mean", "sharedConv.backbone.layer2.0.bn1.running_var", "sharedConv.backbone.layer2.0.conv2.weight", "sharedConv.backbone.layer2.0.bn2.weight", "sharedConv.backbone.layer2.0.bn2.bias", "sharedConv.backbone.layer2.0.bn2.running_mean", "sharedConv.backbone.layer2.0.bn2.running_var", "sharedConv.backbone.layer2.0.conv3.weight", "sharedConv.backbone.layer2.0.bn3.weight", "sharedConv.backbone.layer2.0.bn3.bias", "sharedConv.backbone.layer2.0.bn3.running_mean", "sharedConv.backbone.layer2.0.bn3.running_var", "sharedConv.backbone.layer2.0.downsample.0.weight", "sharedConv.backbone.layer2.0.downsample.1.weight", "sharedConv.backbone.layer2.0.downsample.1.bias", "sharedConv.backbone.layer2.0.downsample.1.running_mean", "sharedConv.backbone.layer2.0.downsample.1.running_var", "sharedConv.backbone.layer2.1.conv1.weight", "sharedConv.backbone.layer2.1.bn1.weight", "sharedConv.backbone.layer2.1.bn1.bias", "sharedConv.backbone.layer2.1.bn1.running_mean", "sharedConv.backbone.layer2.1.bn1.running_var", "sharedConv.backbone.layer2.1.conv2.weight", "sharedConv.backbone.layer2.1.bn2.weight", "sharedConv.backbone.layer2.1.bn2.bias", "sharedConv.backbone.layer2.1.bn2.running_mean", "sharedConv.backbone.layer2.1.bn2.running_var", "sharedConv.backbone.layer2.1.conv3.weight", "sharedConv.backbone.layer2.1.bn3.weight", "sharedConv.backbone.layer2.1.bn3.bias", "sharedConv.backbone.layer2.1.bn3.running_mean", "sharedConv.backbone.layer2.1.bn3.running_var", "sharedConv.backbone.layer2.2.conv1.weight", "sharedConv.backbone.layer2.2.bn1.weight", "sharedConv.backbone.layer2.2.bn1.bias", "sharedConv.backbone.layer2.2.bn1.running_mean", "sharedConv.backbone.layer2.2.bn1.running_var", "sharedConv.backbone.layer2.2.conv2.weight", "sharedConv.backbone.layer2.2.bn2.weight", "sharedConv.backbone.layer2.2.bn2.bias", "sharedConv.backbone.layer2.2.bn2.running_mean", "sharedConv.backbone.layer2.2.bn2.running_var", "sharedConv.backbone.layer2.2.conv3.weight", "sharedConv.backbone.layer2.2.bn3.weight", "sharedConv.backbone.layer2.2.bn3.bias", "sharedConv.backbone.layer2.2.bn3.running_mean", "sharedConv.backbone.layer2.2.bn3.running_var", "sharedConv.backbone.layer2.3.conv1.weight", "sharedConv.backbone.layer2.3.bn1.weight", "sharedConv.backbone.layer2.3.bn1.bias", "sharedConv.backbone.layer2.3.bn1.running_mean", "sharedConv.backbone.layer2.3.bn1.running_var", "sharedConv.backbone.layer2.3.conv2.weight", "sharedConv.backbone.layer2.3.bn2.weight", "sharedConv.backbone.layer2.3.bn2.bias", "sharedConv.backbone.layer2.3.bn2.running_mean", "sharedConv.backbone.layer2.3.bn2.running_var", "sharedConv.backbone.layer2.3.conv3.weight", "sharedConv.backbone.layer2.3.bn3.weight", "sharedConv.backbone.layer2.3.bn3.bias", "sharedConv.backbone.layer2.3.bn3.running_mean", "sharedConv.backbone.layer2.3.bn3.running_var", "sharedConv.backbone.layer3.0.conv1.weight", "sharedConv.backbone.layer3.0.bn1.weight", "sharedConv.backbone.layer3.0.bn1.bias", "sharedConv.backbone.layer3.0.bn1.running_mean", "sharedConv.backbone.layer3.0.bn1.running_var", "sharedConv.backbone.layer3.0.conv2.weight", "sharedConv.backbone.layer3.0.bn2.weight", "sharedConv.backbone.layer3.0.bn2.bias", "sharedConv.backbone.layer3.0.bn2.running_mean", "sharedConv.backbone.layer3.0.bn2.running_var", "sharedConv.backbone.layer3.0.conv3.weight", "sharedConv.backbone.layer3.0.bn3.weight", "sharedConv.backbone.layer3.0.bn3.bias", "sharedConv.backbone.layer3.0.bn3.running_mean", "sharedConv.backbone.layer3.0.bn3.running_var", "sharedConv.backbone.layer3.0.downsample.0.weight", "sharedConv.backbone.layer3.0.downsample.1.weight", "sharedConv.backbone.layer3.0.downsample.1.bias", "sharedConv.backbone.layer3.0.downsample.1.running_mean", "sharedConv.backbone.layer3.0.downsample.1.running_var", "sharedConv.backbone.layer3.1.conv1.weight", "sharedConv.backbone.layer3.1.bn1.weight", "sharedConv.backbone.layer3.1.bn1.bias", "sharedConv.backbone.layer3.1.bn1.running_mean", "sharedConv.backbone.layer3.1.bn1.running_var", "sharedConv.backbone.layer3.1.conv2.weight", "sharedConv.backbone.layer3.1.bn2.weight", "sharedConv.backbone.layer3.1.bn2.bias", "sharedConv.backbone.layer3.1.bn2.running_mean", "sharedConv.backbone.layer3.1.bn2.running_var", "sharedConv.backbone.layer3.1.conv3.weight", "sharedConv.backbone.layer3.1.bn3.weight", "sharedConv.backbone.layer3.1.bn3.bias", "sharedConv.backbone.layer3.1.bn3.running_mean", "sharedConv.backbone.layer3.1.bn3.running_var", "sharedConv.backbone.layer3.2.conv1.weight", "sharedConv.backbone.layer3.2.bn1.weight", "sharedConv.backbone.layer3.2.bn1.bias", "sharedConv.backbone.layer3.2.bn1.running_mean", "sharedConv.backbone.layer3.2.bn1.running_var", "sharedConv.backbone.layer3.2.conv2.weight", "sharedConv.backbone.layer3.2.bn2.weight", "sharedConv.backbone.layer3.2.bn2.bias", "sharedConv.backbone.layer3.2.bn2.running_mean", "sharedConv.backbone.layer3.2.bn2.running_var", "sharedConv.backbone.layer3.2.conv3.weight", "sharedConv.backbone.layer3.2.bn3.weight", "sharedConv.backbone.layer3.2.bn3.bias", "sharedConv.backbone.layer3.2.bn3.running_mean", "sharedConv.backbone.layer3.2.bn3.running_var", "sharedConv.backbone.layer3.3.conv1.weight", "sharedConv.backbone.layer3.3.bn1.weight", "sharedConv.backbone.layer3.3.bn1.bias", "sharedConv.backbone.layer3.3.bn1.running_mean", "sharedConv.backbone.layer3.3.bn1.running_var", "sharedConv.backbone.layer3.3.conv2.weight", "sharedConv.backbone.layer3.3.bn2.weight", "sharedConv.backbone.layer3.3.bn2.bias", "sharedConv.backbone.layer3.3.bn2.running_mean", "sharedConv.backbone.layer3.3.bn2.running_var", "sharedConv.backbone.layer3.3.conv3.weight", "sharedConv.backbone.layer3.3.bn3.weight", "sharedConv.backbone.layer3.3.bn3.bias", "sharedConv.backbone.layer3.3.bn3.running_mean", "sharedConv.backbone.layer3.3.bn3.running_var", "sharedConv.backbone.layer3.4.conv1.weight", "sharedConv.backbone.layer3.4.bn1.weight", "sharedConv.backbone.layer3.4.bn1.bias", "sharedConv.backbone.layer3.4.bn1.running_mean", "sharedConv.backbone.layer3.4.bn1.running_var", "sharedConv.backbone.layer3.4.conv2.weight", "sharedConv.backbone.layer3.4.bn2.weight", "sharedConv.backbone.layer3.4.bn2.bias", "sharedConv.backbone.layer3.4.bn2.running_mean", "sharedConv.backbone.layer3.4.bn2.running_var", "sharedConv.backbone.layer3.4.conv3.weight", "sharedConv.backbone.layer3.4.bn3.weight", "sharedConv.backbone.layer3.4.bn3.bias", "sharedConv.backbone.layer3.4.bn3.running_mean", "sharedConv.backbone.layer3.4.bn3.running_var", "sharedConv.backbone.layer3.5.conv1.weight", "sharedConv.backbone.layer3.5.bn1.weight", "sharedConv.backbone.layer3.5.bn1.bias", "sharedConv.backbone.layer3.5.bn1.running_mean", "sharedConv.backbone.layer3.5.bn1.running_var", "sharedConv.backbone.layer3.5.conv2.weight", "sharedConv.backbone.layer3.5.bn2.weight", "sharedConv.backbone.layer3.5.bn2.bias", "sharedConv.backbone.layer3.5.bn2.running_mean", "sharedConv.backbone.layer3.5.bn2.running_var", "sharedConv.backbone.layer3.5.conv3.weight", "sharedConv.backbone.layer3.5.bn3.weight", "sharedConv.backbone.layer3.5.bn3.bias", "sharedConv.backbone.layer3.5.bn3.running_mean", "sharedConv.backbone.layer3.5.bn3.running_var", "sharedConv.backbone.layer4.0.conv1.weight", "sharedConv.backbone.layer4.0.bn1.weight", "sharedConv.backbone.layer4.0.bn1.bias", "sharedConv.backbone.layer4.0.bn1.running_mean", "sharedConv.backbone.layer4.0.bn1.running_var", "sharedConv.backbone.layer4.0.conv2.weight", "sharedConv.backbone.layer4.0.bn2.weight", "sharedConv.backbone.layer4.0.bn2.bias", "sharedConv.backbone.layer4.0.bn2.running_mean", "sharedConv.backbone.layer4.0.bn2.running_var", "sharedConv.backbone.layer4.0.conv3.weight", "sharedConv.backbone.layer4.0.bn3.weight", "sharedConv.backbone.layer4.0.bn3.bias", "sharedConv.backbone.layer4.0.bn3.running_mean", "sharedConv.backbone.layer4.0.bn3.running_var", "sharedConv.backbone.layer4.0.downsample.0.weight", "sharedConv.backbone.layer4.0.downsample.1.weight", "sharedConv.backbone.layer4.0.downsample.1.bias", "sharedConv.backbone.layer4.0.downsample.1.running_mean", "sharedConv.backbone.layer4.0.downsample.1.running_var", "sharedConv.backbone.layer4.1.conv1.weight", "sharedConv.backbone.layer4.1.bn1.weight", "sharedConv.backbone.layer4.1.bn1.bias", "sharedConv.backbone.layer4.1.bn1.running_mean", "sharedConv.backbone.layer4.1.bn1.running_var", "sharedConv.backbone.layer4.1.conv2.weight", "sharedConv.backbone.layer4.1.bn2.weight", "sharedConv.backbone.layer4.1.bn2.bias", "sharedConv.backbone.layer4.1.bn2.running_mean", "sharedConv.backbone.layer4.1.bn2.running_var", "sharedConv.backbone.layer4.1.conv3.weight", "sharedConv.backbone.layer4.1.bn3.weight", "sharedConv.backbone.layer4.1.bn3.bias", "sharedConv.backbone.layer4.1.bn3.running_mean", "sharedConv.backbone.layer4.1.bn3.running_var", "sharedConv.backbone.layer4.2.conv1.weight", "sharedConv.backbone.layer4.2.bn1.weight", "sharedConv.backbone.layer4.2.bn1.bias", "sharedConv.backbone.layer4.2.bn1.running_mean", "sharedConv.backbone.layer4.2.bn1.running_var", "sharedConv.backbone.layer4.2.conv2.weight", "sharedConv.backbone.layer4.2.bn2.weight", "sharedConv.backbone.layer4.2.bn2.bias", "sharedConv.backbone.layer4.2.bn2.running_mean", "sharedConv.backbone.layer4.2.bn2.running_var", "sharedConv.backbone.layer4.2.conv3.weight", "sharedConv.backbone.layer4.2.bn3.weight", "sharedConv.backbone.layer4.2.bn3.bias", "sharedConv.backbone.layer4.2.bn3.running_mean", "sharedConv.backbone.layer4.2.bn3.running_var", "sharedConv.backbone.last_linear.weight", "sharedConv.backbone.last_linear.bias", "sharedConv.mergeLayers1.conv2dOne.weight", "sharedConv.mergeLayers1.conv2dOne.bias", "sharedConv.mergeLayers1.bnOne.weight", "sharedConv.mergeLayers1.bnOne.bias", "sharedConv.mergeLayers1.bnOne.running_mean", "sharedConv.mergeLayers1.bnOne.running_var", "sharedConv.mergeLayers1.conv2dTwo.weight", "sharedConv.mergeLayers1.conv2dTwo.bias", "sharedConv.mergeLayers1.bnTwo.weight", "sharedConv.mergeLayers1.bnTwo.bias", "sharedConv.mergeLayers1.bnTwo.running_mean", "sharedConv.mergeLayers1.bnTwo.running_var", "sharedConv.mergeLayers2.conv2dOne.weight", "sharedConv.mergeLayers2.conv2dOne.bias", "sharedConv.mergeLayers2.bnOne.weight", "sharedConv.mergeLayers2.bnOne.bias", "sharedConv.mergeLayers2.bnOne.running_mean", "sharedConv.mergeLayers2.bnOne.running_var", "sharedConv.mergeLayers2.conv2dTwo.weight", "sharedConv.mergeLayers2.conv2dTwo.bias", "sharedConv.mergeLayers2.bnTwo.weight", "sharedConv.mergeLayers2.bnTwo.bias", "sharedConv.mergeLayers2.bnTwo.running_mean", "sharedConv.mergeLayers2.bnTwo.running_var", "sharedConv.mergeLayers3.conv2dOne.weight", "sharedConv.mergeLayers3.conv2dOne.bias", "sharedConv.mergeLayers3.bnOne.weight", "sharedConv.mergeLayers3.bnOne.bias", "sharedConv.mergeLayers3.bnOne.running_mean", "sharedConv.mergeLayers3.bnOne.running_var", "sharedConv.mergeLayers3.conv2dTwo.weight", "sharedConv.mergeLayers3.conv2dTwo.bias", "sharedConv.mergeLayers3.bnTwo.weight", "sharedConv.mergeLayers3.bnTwo.bias", "sharedConv.mergeLayers3.bnTwo.running_mean", "sharedConv.mergeLayers3.bnTwo.running_var", "sharedConv.mergeLayers4.weight", "sharedConv.mergeLayers4.bias", "sharedConv.bn5.weight", "sharedConv.bn5.bias", "sharedConv.bn5.running_mean", "sharedConv.bn5.running_var", "sharedConv.scoreMap.weight", "sharedConv.scoreMap.bias", "sharedConv.geoMap.weight", "sharedConv.geoMap.bias", "sharedConv.angleMap.weight", "sharedConv.angleMap.bias". Unexpected key(s) in state_dict: "module.sharedConv.backbone.conv1.weight", "module.sharedConv.backbone.bn1.weight", "module.sharedConv.backbone.bn1.bias", "module.sharedConv.backbone.bn1.running_mean", "module.sharedConv.backbone.bn1.running_var", "module.sharedConv.backbone.bn1.num_batches_tracked", "module.sharedConv.backbone.layer1.0.conv1.weight", "module.sharedConv.backbone.layer1.0.bn1.weight", "module.sharedConv.backbone.layer1.0.bn1.bias", "module.sharedConv.backbone.layer1.0.bn1.running_mean", "module.sharedConv.backbone.layer1.0.bn1.running_var", "module.sharedConv.backbone.layer1.0.bn1.num_batches_tracked", "module.sharedConv.backbone.layer1.0.conv2.weight", "module.sharedConv.backbone.layer1.0.bn2.weight", "module.sharedConv.backbone.layer1.0.bn2.bias", "module.sharedConv.backbone.layer1.0.bn2.running_mean", "module.sharedConv.backbone.layer1.0.bn2.running_var", "module.sharedConv.backbone.layer1.0.bn2.num_batches_tracked", "module.sharedConv.backbone.layer1.0.conv3.weight", "module.sharedConv.backbone.layer1.0.bn3.weight", "module.sharedConv.backbone.layer1.0.bn3.bias", "module.sharedConv.backbone.layer1.0.bn3.running_mean", "module.sharedConv.backbone.layer1.0.bn3.running_var", "module.sharedConv.backbone.layer1.0.bn3.num_batches_tracked", "module.sharedConv.backbone.layer1.0.downsample.0.weight", "module.sharedConv.backbone.layer1.0.downsample.1.weight", "module.sharedConv.backbone.layer1.0.downsample.1.bias", "module.sharedConv.backbone.layer1.0.downsample.1.running_mean", "module.sharedConv.backbone.layer1.0.downsample.1.running_var", "module.sharedConv.backbone.layer1.0.downsample.1.num_batches_tracked", "module.sharedConv.backbone.layer1.1.conv1.weight", "module.sharedConv.backbone.layer1.1.bn1.weight", "module.sharedConv.backbone.layer1.1.bn1.bias", "module.sharedConv.backbone.layer1.1.bn1.running_mean", "module.sharedConv.backbone.layer1.1.bn1.running_var", "module.sharedConv.backbone.layer1.1.bn1.num_batches_tracked", "module.sharedConv.backbone.layer1.1.conv2.weight", "module.sharedConv.backbone.layer1.1.bn2.weight", "module.sharedConv.backbone.layer1.1.bn2.bias", "module.sharedConv.backbone.layer1.1.bn2.running_mean", "module.sharedConv.backbone.layer1.1.bn2.running_var", "module.sharedConv.backbone.layer1.1.bn2.num_batches_tracked", "module.sharedConv.backbone.layer1.1.conv3.weight", "module.sharedConv.backbone.layer1.1.bn3.weight", "module.sharedConv.backbone.layer1.1.bn3.bias", "module.sharedConv.backbone.layer1.1.bn3.running_mean", "module.sharedConv.backbone.layer1.1.bn3.running_var", "module.sharedConv.backbone.layer1.1.bn3.num_batches_tracked", "module.sharedConv.backbone.layer1.2.conv1.weight", "module.sharedConv.backbone.layer1.2.bn1.weight", "module.sharedConv.backbone.layer1.2.bn1.bias", "module.sharedConv.backbone.layer1.2.bn1.running_mean", "module.sharedConv.backbone.layer1.2.bn1.running_var", "module.sharedConv.backbone.layer1.2.bn1.num_batches_tracked", "module.sharedConv.backbone.layer1.2.conv2.weight", "module.sharedConv.backbone.layer1.2.bn2.weight", "module.sharedConv.backbone.layer1.2.bn2.bias", "module.sharedConv.backbone.layer1.2.bn2.running_mean", "module.sharedConv.backbone.layer1.2.bn2.running_var", "module.sharedConv.backbone.layer1.2.bn2.num_batches_tracked", "module.sharedConv.backbone.layer1.2.conv3.weight", "module.sharedConv.backbone.layer1.2.bn3.weight", "module.sharedConv.backbone.layer1.2.bn3.bias", "module.sharedConv.backbone.layer1.2.bn3.running_mean", "module.sharedConv.backbone.layer1.2.bn3.running_var", "module.sharedConv.backbone.layer1.2.bn3.num_batches_tracked", "module.sharedConv.backbone.layer2.0.conv1.weight", "module.sharedConv.backbone.layer2.0.bn1.weight", "module.sharedConv.backbone.layer2.0.bn1.bias", "module.sharedConv.backbone.layer2.0.bn1.running_mean", "module.sharedConv.backbone.layer2.0.bn1.running_var", "module.sharedConv.backbone.layer2.0.bn1.num_batches_tracked", "module.sharedConv.backbone.layer2.0.conv2.weight", "module.sharedConv.backbone.layer2.0.bn2.weight", "module.sharedConv.backbone.layer2.0.bn2.bias", "module.sharedConv.backbone.layer2.0.bn2.running_mean", "module.sharedConv.backbone.layer2.0.bn2.running_var", "module.sharedConv.backbone.layer2.0.bn2.num_batches_tracked", "module.sharedConv.backbone.layer2.0.conv3.weight", "module.sharedConv.backbone.layer2.0.bn3.weight", "module.sharedConv.backbone.layer2.0.bn3.bias", "module.sharedConv.backbone.layer2.0.bn3.running_mean", "module.sharedConv.backbone.layer2.0.bn3.running_var", "module.sharedConv.backbone.layer2.0.bn3.num_batches_tracked", "module.sharedConv.backbone.layer2.0.downsample.0.weight", "module.sharedConv.backbone.layer2.0.downsample.1.weight", "module.sharedConv.backbone.layer2.0.downsample.1.bias", "module.sharedConv.backbone.layer2.0.downsample.1.running_mean", "module.sharedConv.backbone.layer2.0.downsample.1.running_var", "module.sharedConv.backbone.layer2.0.downsample.1.num_batches_tracked", "module.sharedConv.backbone.layer2.1.conv1.weight", "module.sharedConv.backbone.layer2.1.bn1.weight", "module.sharedConv.backbone.layer2.1.bn1.bias", "module.sharedConv.backbone.layer2.1.bn1.running_mean", "module.sharedConv.backbone.layer2.1.bn1.running_var", "module.sharedConv.backbone.layer2.1.bn1.num_batches_tracked", "module.sharedConv.backbone.layer2.1.conv2.weight", "module.sharedConv.backbone.layer2.1.bn2.weight", "module.sharedConv.backbone.layer2.1.bn2.bias", "module.sharedConv.backbone.layer2.1.bn2.running_mean", "module.sharedConv.backbone.layer2.1.bn2.running_var", "module.sharedConv.backbone.layer2.1.bn2.num_batches_tracked", "module.sharedConv.backbone.layer2.1.conv3.weight", "module.sharedConv.backbone.layer2.1.bn3.weight", "module.sharedConv.backbone.layer2.1.bn3.bias", "module.sharedConv.backbone.layer2.1.bn3.running_mean", "module.sharedConv.backbone.layer2.1.bn3.running_var", "module.sharedConv.backbone.layer2.1.bn3.num_batches_tracked", "module.sharedConv.backbone.layer2.2.conv1.weight", "module.sharedConv.backbone.layer2.2.bn1.weight", "module.sharedConv.backbone.layer2.2.bn1.bias", "module.sharedConv.backbone.layer2.2.bn1.running_mean", "module.sharedConv.backbone.layer2.2.bn1.running_var", "module.sharedConv.backbone.layer2.2.bn1.num_batches_tracked", "module.sharedConv.backbone.layer2.2.conv2.weight", "module.sharedConv.backbone.layer2.2.bn2.weight", "module.sharedConv.backbone.layer2.2.bn2.bias", "module.sharedConv.backbone.layer2.2.bn2.running_mean", "module.sharedConv.backbone.layer2.2.bn2.running_var", "module.sharedConv.backbone.layer2.2.bn2.num_batches_tracked", "module.sharedConv.backbone.layer2.2.conv3.weight", "module.sharedConv.backbone.layer2.2.bn3.weight", "module.sharedConv.backbone.layer2.2.bn3.bias", "module.sharedConv.backbone.layer2.2.bn3.running_mean", "module.sharedConv.backbone.layer2.2.bn3.running_var", "module.sharedConv.backbone.layer2.2.bn3.num_batches_tracked", "module.sharedConv.backbone.layer2.3.conv1.weight", "module.sharedConv.backbone.layer2.3.bn1.weight", "module.sharedConv.backbone.layer2.3.bn1.bias", "module.sharedConv.backbone.layer2.3.bn1.running_mean", "module.sharedConv.backbone.layer2.3.bn1.running_var", "module.sharedConv.backbone.layer2.3.bn1.num_batches_tracked", "module.sharedConv.backbone.layer2.3.conv2.weight", "module.sharedConv.backbone.layer2.3.bn2.weight", "module.sharedConv.backbone.layer2.3.bn2.bias", "module.sharedConv.backbone.layer2.3.bn2.running_mean", "module.sharedConv.backbone.layer2.3.bn2.running_var", "module.sharedConv.backbone.layer2.3.bn2.num_batches_tracked", "module.sharedConv.backbone.layer2.3.conv3.weight", "module.sharedConv.backbone.layer2.3.bn3.weight", "module.sharedConv.backbone.layer2.3.bn3.bias", "module.sharedConv.backbone.layer2.3.bn3.running_mean", "module.sharedConv.backbone.layer2.3.bn3.running_var", "module.sharedConv.backbone.layer2.3.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.0.conv1.weight", "module.sharedConv.backbone.layer3.0.bn1.weight", "module.sharedConv.backbone.layer3.0.bn1.bias", "module.sharedConv.backbone.layer3.0.bn1.running_mean", "module.sharedConv.backbone.layer3.0.bn1.running_var", "module.sharedConv.backbone.layer3.0.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.0.conv2.weight", "module.sharedConv.backbone.layer3.0.bn2.weight", "module.sharedConv.backbone.layer3.0.bn2.bias", "module.sharedConv.backbone.layer3.0.bn2.running_mean", "module.sharedConv.backbone.layer3.0.bn2.running_var", "module.sharedConv.backbone.layer3.0.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.0.conv3.weight", "module.sharedConv.backbone.layer3.0.bn3.weight", "module.sharedConv.backbone.layer3.0.bn3.bias", "module.sharedConv.backbone.layer3.0.bn3.running_mean", "module.sharedConv.backbone.layer3.0.bn3.running_var", "module.sharedConv.backbone.layer3.0.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.0.downsample.0.weight", "module.sharedConv.backbone.layer3.0.downsample.1.weight", "module.sharedConv.backbone.layer3.0.downsample.1.bias", "module.sharedConv.backbone.layer3.0.downsample.1.running_mean", "module.sharedConv.backbone.layer3.0.downsample.1.running_var", "module.sharedConv.backbone.layer3.0.downsample.1.num_batches_tracked", "module.sharedConv.backbone.layer3.1.conv1.weight", "module.sharedConv.backbone.layer3.1.bn1.weight", "module.sharedConv.backbone.layer3.1.bn1.bias", "module.sharedConv.backbone.layer3.1.bn1.running_mean", "module.sharedConv.backbone.layer3.1.bn1.running_var", "module.sharedConv.backbone.layer3.1.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.1.conv2.weight", "module.sharedConv.backbone.layer3.1.bn2.weight", "module.sharedConv.backbone.layer3.1.bn2.bias", "module.sharedConv.backbone.layer3.1.bn2.running_mean", "module.sharedConv.backbone.layer3.1.bn2.running_var", "module.sharedConv.backbone.layer3.1.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.1.conv3.weight", "module.sharedConv.backbone.layer3.1.bn3.weight", "module.sharedConv.backbone.layer3.1.bn3.bias", "module.sharedConv.backbone.layer3.1.bn3.running_mean", "module.sharedConv.backbone.layer3.1.bn3.running_var", "module.sharedConv.backbone.layer3.1.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.2.conv1.weight", "module.sharedConv.backbone.layer3.2.bn1.weight", "module.sharedConv.backbone.layer3.2.bn1.bias", "module.sharedConv.backbone.layer3.2.bn1.running_mean", "module.sharedConv.backbone.layer3.2.bn1.running_var", "module.sharedConv.backbone.layer3.2.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.2.conv2.weight", "module.sharedConv.backbone.layer3.2.bn2.weight", "module.sharedConv.backbone.layer3.2.bn2.bias", "module.sharedConv.backbone.layer3.2.bn2.running_mean", "module.sharedConv.backbone.layer3.2.bn2.running_var", "module.sharedConv.backbone.layer3.2.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.2.conv3.weight", "module.sharedConv.backbone.layer3.2.bn3.weight", "module.sharedConv.backbone.layer3.2.bn3.bias", "module.sharedConv.backbone.layer3.2.bn3.running_mean", "module.sharedConv.backbone.layer3.2.bn3.running_var", "module.sharedConv.backbone.layer3.2.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.3.conv1.weight", "module.sharedConv.backbone.layer3.3.bn1.weight", "module.sharedConv.backbone.layer3.3.bn1.bias", "module.sharedConv.backbone.layer3.3.bn1.running_mean", "module.sharedConv.backbone.layer3.3.bn1.running_var", "module.sharedConv.backbone.layer3.3.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.3.conv2.weight", "module.sharedConv.backbone.layer3.3.bn2.weight", "module.sharedConv.backbone.layer3.3.bn2.bias", "module.sharedConv.backbone.layer3.3.bn2.running_mean", "module.sharedConv.backbone.layer3.3.bn2.running_var", "module.sharedConv.backbone.layer3.3.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.3.conv3.weight", "module.sharedConv.backbone.layer3.3.bn3.weight", "module.sharedConv.backbone.layer3.3.bn3.bias", "module.sharedConv.backbone.layer3.3.bn3.running_mean", "module.sharedConv.backbone.layer3.3.bn3.running_var", "module.sharedConv.backbone.layer3.3.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.4.conv1.weight", "module.sharedConv.backbone.layer3.4.bn1.weight", "module.sharedConv.backbone.layer3.4.bn1.bias", "module.sharedConv.backbone.layer3.4.bn1.running_mean", "module.sharedConv.backbone.layer3.4.bn1.running_var", "module.sharedConv.backbone.layer3.4.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.4.conv2.weight", "module.sharedConv.backbone.layer3.4.bn2.weight", "module.sharedConv.backbone.layer3.4.bn2.bias", "module.sharedConv.backbone.layer3.4.bn2.running_mean", "module.sharedConv.backbone.layer3.4.bn2.running_var", "module.sharedConv.backbone.layer3.4.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.4.conv3.weight", "module.sharedConv.backbone.layer3.4.bn3.weight", "module.sharedConv.backbone.layer3.4.bn3.bias", "module.sharedConv.backbone.layer3.4.bn3.running_mean", "module.sharedConv.backbone.layer3.4.bn3.running_var", "module.sharedConv.backbone.layer3.4.bn3.num_batches_tracked", "module.sharedConv.backbone.layer3.5.conv1.weight", "module.sharedConv.backbone.layer3.5.bn1.weight", "module.sharedConv.backbone.layer3.5.bn1.bias", "module.sharedConv.backbone.layer3.5.bn1.running_mean", "module.sharedConv.backbone.layer3.5.bn1.running_var", "module.sharedConv.backbone.layer3.5.bn1.num_batches_tracked", "module.sharedConv.backbone.layer3.5.conv2.weight", "module.sharedConv.backbone.layer3.5.bn2.weight", "module.sharedConv.backbone.layer3.5.bn2.bias", "module.sharedConv.backbone.layer3.5.bn2.running_mean", "module.sharedConv.backbone.layer3.5.bn2.running_var", "module.sharedConv.backbone.layer3.5.bn2.num_batches_tracked", "module.sharedConv.backbone.layer3.5.conv3.weight", "module.sharedConv.backbone.layer3.5.bn3.weight", "module.sharedConv.backbone.layer3.5.bn3.bias", "module.sharedConv.backbone.layer3.5.bn3.running_mean", "module.sharedConv.backbone.layer3.5.bn3.running_var", "module.sharedConv.backbone.layer3.5.bn3.num_batches_tracked", "module.sharedConv.backbone.layer4.0.conv1.weight", "module.sharedConv.backbone.layer4.0.bn1.weight", "module.sharedConv.backbone.layer4.0.bn1.bias", "module.sharedConv.backbone.layer4.0.bn1.running_mean", "module.sharedConv.backbone.layer4.0.bn1.running_var", "module.sharedConv.backbone.layer4.0.bn1.num_batches_tracked", "module.sharedConv.backbone.layer4.0.conv2.weight", "module.sharedConv.backbone.layer4.0.bn2.weight", "module.sharedConv.backbone.layer4.0.bn2.bias", "module.sharedConv.backbone.layer4.0.bn2.running_mean", "module.sharedConv.backbone.layer4.0.bn2.running_var", "module.sharedConv.backbone.layer4.0.bn2.num_batches_tracked", "module.sharedConv.backbone.layer4.0.conv3.weight", "module.sharedConv.backbone.layer4.0.bn3.weight", "module.sharedConv.backbone.layer4.0.bn3.bias", "module.sharedConv.backbone.layer4.0.bn3.running_mean", "module.sharedConv.backbone.layer4.0.bn3.running_var", "module.sharedConv.backbone.layer4.0.bn3.num_batches_tracked", "module.sharedConv.backbone.layer4.0.downsample.0.weight", "module.sharedConv.backbone.layer4.0.downsample.1.weight", "module.sharedConv.backbone.layer4.0.downsample.1.bias", "module.sharedConv.backbone.layer4.0.downsample.1.running_mean", "module.sharedConv.backbone.layer4.0.downsample.1.running_var", "module.sharedConv.backbone.layer4.0.downsample.1.num_batches_tracked", "module.sharedConv.backbone.layer4.1.conv1.weight", "module.sharedConv.backbone.layer4.1.bn1.weight", "module.sharedConv.backbone.layer4.1.bn1.bias", "module.sharedConv.backbone.layer4.1.bn1.running_mean", "module.sharedConv.backbone.layer4.1.bn1.running_var", "module.sharedConv.backbone.layer4.1.bn1.num_batches_tracked", "module.sharedConv.backbone.layer4.1.conv2.weight", "module.sharedConv.backbone.layer4.1.bn2.weight", "module.sharedConv.backbone.layer4.1.bn2.bias", "module.sharedConv.backbone.layer4.1.bn2.running_mean", "module.sharedConv.backbone.layer4.1.bn2.running_var", "module.sharedConv.backbone.layer4.1.bn2.num_batches_tracked", "module.sharedConv.backbone.layer4.1.conv3.weight", "module.sharedConv.backbone.layer4.1.bn3.weight", "module.sharedConv.backbone.layer4.1.bn3.bias", "module.sharedConv.backbone.layer4.1.bn3.running_mean", "module.sharedConv.backbone.layer4.1.bn3.running_var", "module.sharedConv.backbone.layer4.1.bn3.num_batches_tracked", "module.sharedConv.backbone.layer4.2.conv1.weight", "module.sharedConv.backbone.layer4.2.bn1.weight", "module.sharedConv.backbone.layer4.2.bn1.bias", "module.sharedConv.backbone.layer4.2.bn1.running_mean", "module.sharedConv.backbone.layer4.2.bn1.running_var", "module.sharedConv.backbone.layer4.2.bn1.num_batches_tracked", "module.sharedConv.backbone.layer4.2.conv2.weight", "module.sharedConv.backbone.layer4.2.bn2.weight", "module.sharedConv.backbone.layer4.2.bn2.bias", "module.sharedConv.backbone.layer4.2.bn2.running_mean", "module.sharedConv.backbone.layer4.2.bn2.running_var", "module.sharedConv.backbone.layer4.2.bn2.num_batches_tracked", "module.sharedConv.backbone.layer4.2.conv3.weight", "module.sharedConv.backbone.layer4.2.bn3.weight", "module.sharedConv.backbone.layer4.2.bn3.bias", "module.sharedConv.backbone.layer4.2.bn3.running_mean", "module.sharedConv.backbone.layer4.2.bn3.running_var", "module.sharedConv.backbone.layer4.2.bn3.num_batches_tracked", "module.sharedConv.backbone.last_linear.weight", "module.sharedConv.backbone.last_linear.bias", "module.sharedConv.mergeLayers1.conv2dOne.weight", "module.sharedConv.mergeLayers1.conv2dOne.bias", "module.sharedConv.mergeLayers1.bnOne.weight", "module.sharedConv.mergeLayers1.bnOne.bias", "module.sharedConv.mergeLayers1.bnOne.running_mean", "module.sharedConv.mergeLayers1.bnOne.running_var", "module.sharedConv.mergeLayers1.bnOne.num_batches_tracked", "module.sharedConv.mergeLayers1.conv2dTwo.weight", "module.sharedConv.mergeLayers1.conv2dTwo.bias", "module.sharedConv.mergeLayers1.bnTwo.weight", "module.sharedConv.mergeLayers1.bnTwo.bias", "module.sharedConv.mergeLayers1.bnTwo.running_mean", "module.sharedConv.mergeLayers1.bnTwo.running_var", "module.sharedConv.mergeLayers1.bnTwo.num_batches_tracked", "module.sharedConv.mergeLayers2.conv2dOne.weight", "module.sharedConv.mergeLayers2.conv2dOne.bias", "module.sharedConv.mergeLayers2.bnOne.weight", "module.sharedConv.mergeLayers2.bnOne.bias", "module.sharedConv.mergeLayers2.bnOne.running_mean", "module.sharedConv.mergeLayers2.bnOne.running_var", "module.sharedConv.mergeLayers2.bnOne.num_batches_tracked", "module.sharedConv.mergeLayers2.conv2dTwo.weight", "module.sharedConv.mergeLayers2.conv2dTwo.bias", "module.sharedConv.mergeLayers2.bnTwo.weight", "module.sharedConv.mergeLayers2.bnTwo.bias", "module.sharedConv.mergeLayers2.bnTwo.running_mean", "module.sharedConv.mergeLayers2.bnTwo.running_var", "module.sharedConv.mergeLayers2.bnTwo.num_batches_tracked", "module.sharedConv.mergeLayers3.conv2dOne.weight", "module.sharedConv.mergeLayers3.conv2dOne.bias", "module.sharedConv.mergeLayers3.bnOne.weight", "module.sharedConv.mergeLayers3.bnOne.bias", "module.sharedConv.mergeLayers3.bnOne.running_mean", "module.sharedConv.mergeLayers3.bnOne.running_var", "module.sharedConv.mergeLayers3.bnOne.num_batches_tracked", "module.sharedConv.mergeLayers3.conv2dTwo.weight", "module.sharedConv.mergeLayers3.conv2dTwo.bias", "module.sharedConv.mergeLayers3.bnTwo.weight", "module.sharedConv.mergeLayers3.bnTwo.bias", "module.sharedConv.mergeLayers3.bnTwo.running_mean", "module.sharedConv.mergeLayers3.bnTwo.running_var", "module.sharedConv.mergeLayers3.bnTwo.num_batches_tracked", "module.sharedConv.mergeLayers4.weight", "module.sharedConv.mergeLayers4.bias", "module.sharedConv.bn5.weight", "module.sharedConv.bn5.bias", "module.sharedConv.bn5.running_mean", "module.sharedConv.bn5.running_var", "module.sharedConv.bn5.num_batches_tracked", "module.sharedConv.scoreMap.weight", "module.sharedConv.scoreMap.bias", "module.sharedConv.geoMap.weight", "module.sharedConv.geoMap.bias", "module.sharedConv.angleMap.weight", "module.sharedConv.angleMap.bias".

donglin8506 commented 5 years ago

这个问题是跟多GPU训练相关，如果是单块GPU训练的模型，需要将eval.py 20行附近的 # model = torch.nn.DataParallel(model)注释；如果是多块GPU训练，则需要去掉该行的注释。

ntdat017 commented 5 years ago

I have same problem and solve it by comment line 20 in eval.py. =))