hellozhuo / pidinet

Code for the ICCV 2021 paper "Pixel Difference Networks for Efficient Edge Detection" (Oral).
Other
445 stars 69 forks source link

请问这是什么问题呢,我在用自己训练得到的模型进行验证的时候出现的 #47

Closed Rainbow0Dash closed 1 year ago

Rainbow0Dash commented 1 year ago

raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.attentions.0.conv1.weight", "module.attentions.0.conv1.bias", "module.attentions.0.conv2.weight", "module.attentions.1.conv1.weight", "module.attentions.1.conv1.bias", "module.attentions.1.conv2.weight" , "module.attentions.2.conv1.weight", "module.attentions.2.conv1.bias", "module.attentions.2.conv2.weight", "module.attentions.3.conv1.weight", "module.attentions.3.conv1.bias", "module.attentions.3.conv2.weight", "module.dilations.0.conv1.weight", "module.dilations.0.conv1.bias", "module.dilations.0.conv2_1.weight", "module.dilations.0.conv2_2.weight", "module.dilations.0.conv2_3.weight", "module.dilations.0.conv2_4.weight", "module.dilations.1.conv1.weight", "module.dilations.1.conv1.bias" , "module.dilations.1.conv2_1.weight", "module.dilations.1.conv2_2.weight", "module.dilations.1.conv2_3.weight", "module.dilations.1.conv2_4.weight", "module.dilations.2.conv1.weight", "module.dilations.2.conv1.bias", "module.dilations.2.conv2_1.we ight", "module.dilations.2.conv2_2.weight", "module.dilations.2.conv2_3.weight", "module.dilations.2.conv2_4.weight", "module.dilations.3.conv1.weight", "module.dilations.3.conv1.bias", "module.dilations.3.conv2_1.weight", "module.dilations.3.conv2 _2.weight", "module.dilations.3.conv2_3.weight", "module.dilations.3.conv2_4.weight". size mismatch for module.conv_reduces.0.conv.weight: copying a param with shape torch.Size([1, 60, 1, 1]) from checkpoint, the shape in current model is torch.Size([1, 24, 1, 1]). size mismatch for module.conv_reduces.1.conv.weight: copying a param with shape torch.Size([1, 120, 1, 1]) from checkpoint, the shape in current model is torch.Size([1, 24, 1, 1]). size mismatch for module.conv_reduces.2.conv.weight: copying a param with shape torch.Size([1, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([1, 24, 1, 1]). size mismatch for module.conv_reduces.3.conv.weight: copying a param with shape torch.Size([1, 240, 1, 1]) from checkpoint, the shape in current model is torch.Size([1, 24, 1, 1]).

Rainbow0Dash commented 1 year ago

多了个参数,找到问题所在了