最近找到了pytorch版本的pspnet,但是運行的時候卻報了下面的錯,折騰了好久,原來是pytorch版本的原因,
pip3 uninstall torch
pip3 install torch==0.3.1
即可解決問題.
RuntimeError: Error(s) in loading state_dict for ResNet:
Missing key(s) in state_dict: "layer4.0.bn2.running_var", "layer4.0.bn2.running_mean", "layer4.0.conv3.weight", "layer4.0.bn3.weight", "layer4.0.bn3.running_var", "layer4.0.bn3.bias", "layer4.0.bn3.running_mean", "layer4.0.downsample.0.weight", "layer4.0.downsample.1.weight", "layer4.0.downsample.1.running_var", "layer4.0.downsample.1.bias", "layer4.0.downsample.1.running_mean", "layer4.1.conv1.weight", "layer4.1.bn1.weight", "layer4.1.bn1.running_var", "layer4.1.bn1.bias", "layer4.1.bn1.running_mean", "layer4.1.conv2.weight", "layer4.1.bn2.weight", "layer4.1.bn2.running_var", "layer4.1.bn2.bias", "layer4.1.bn2.running_mean", "layer4.1.conv3.weight", "layer4.1.bn3.weight", "layer4.1.bn3.running_var", "layer4.1.bn3.bias", "layer4.1.bn3.running_mean", "layer4.2.conv1.weight", "layer4.2.bn1.weight", "layer4.2.bn1.running_var", "layer4.2.bn1.bias", "layer4.2.bn1.running_mean", "layer4.2.conv2.weight", "layer4.2.bn2.weight", "layer4.2.bn2.running_var", "layer4.2.bn2.bias", "layer4.2.bn2.running_mean", "layer4.2.conv3.weight", "layer4.2.bn3.weight", "layer4.2.bn3.running_var", "layer4.2.bn3.bias", "layer4.2.bn3.running_mean".
size mismatch for bn1.num_batches_tracked: copying a param of torch.Size([]) from checkpoint, where the shape is torch.Size([64, 64, 1, 1]) in current model.
size mismatch for layer1.0.conv1.weight: copying a param of torch.Size([64, 64, 1, 1]) from checkpoint, where the shape is torch.Size([64]) in current model.