DBnet-lite.pytorch icon indicating copy to clipboard operation
DBnet-lite.pytorch copied to clipboard

RuntimeError: Error(s) in loading state_dict for DBNet:

Open the-star-sea opened this issue 3 years ago • 1 comments
trafficstars

when I run train_finetune,there is an error

Resuming from checkpoint.
Traceback (most recent call last):
  File "/home/DISCOVER_summer2022/zhangt/dblite/pruned/train_fintune.py", line 212, in <module>
    train_net(config)
  File "/home/DISCOVER_summer2022/zhangt/dblite/pruned/train_fintune.py", line 90, in train_net
    model.load_state_dict(checkpoint['state_dict'])
  File "/home/DISCOVER_summer2022/zhangt/.conda/envs/dbnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1625, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DBNet:
	Missing key(s) in state_dict: "backbone.conv1.weight", "backbone.bn1.weight", "backbone.bn1.bias", "backbone.bn1.running_mean", "backbone.bn1.running_var", "backbone.layer1.0.conv1.weight", "backbone.layer1.0.bn1.weight", "backbone.layer1.0.bn1.bias", "backbone.layer1.0.bn1.running_mean", "backbone.layer1.0.bn1.running_var", "backbone.layer1.0.conv2.weight", "backbone.layer1.0.bn2.weight", "backbone.layer1.0.bn2.bias", "backbone.layer1.0.bn2.running_mean", "backbone.layer1.0.bn2.running_var", "backbone.layer1.1.conv1.weight", "backbone.layer1.1.bn1.weight", "backbone.layer1.1.bn1.bias", "backbone.layer1.1.bn1.running_mean", "backbone.layer1.1.bn1.running_var", "backbone.layer1.1.conv2.weight", "backbone.layer1.1.bn2.weight", "backbone.layer1.1.bn2.bias", "backbone.layer1.1.bn2.running_mean", "backbone.layer1.1.bn2.running_var", "backbone.layer2.0.conv1.weight", "backbone.layer2.0.bn1.weight", "backbone.layer2.0.bn1.bias", "backbone.layer2.0.bn1.running_mean", "backbone.layer2.0.bn1.running_var", "backbone.layer2.0.conv2.weight", "backbone.layer2.0.bn2.weight", "backbone.layer2.0.bn2.bias", "backbone.layer2.0.bn2.running_mean", "backbone.layer2.0.bn2.running_var", "backbone.layer2.0.downsample.0.weight", "backbone.layer2.0.downsample.1.weight", "backbone.layer2.0.downsample.1.bias", "backbone.layer2.0.downsample.1.running_mean", "backbone.layer2.0.downsample.1.running_var", "backbone.layer2.1.conv1.weight", "backbone.layer2.1.bn1.weight", "backbone.layer2.1.bn1.bias", "backbone.layer2.1.bn1.running_mean", "backbone.layer2.1.bn1.running_var", "backbone.layer2.1.conv2.weight", "backbone.layer2.1.bn2.weight", "backbone.layer2.1.bn2.bias", "backbone.layer2.1.bn2.running_mean", "backbone.layer2.1.bn2.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.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.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.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", "decode.head.in5.conv.weight", "decode.head.in5.bn.weight", "decode.head.in5.bn.bias", "decode.head.in5.bn.running_mean", "decode.head.in5.bn.running_var", "decode.head.in4.conv.weight", "decode.head.in4.bn.weight", "decode.head.in4.bn.bias", "decode.head.in4.bn.running_mean", "decode.head.in4.bn.running_var", "decode.head.in3.conv.weight", "decode.head.in3.bn.weight", "decode.head.in3.bn.bias", "decode.head.in3.bn.running_mean", "decode.head.in3.bn.running_var", "decode.head.in2.conv.weight", "decode.head.in2.bn.weight", "decode.head.in2.bn.bias", "decode.head.in2.bn.running_mean", "decode.head.in2.bn.running_var", "decode.head.out5.0.conv.weight", "decode.head.out5.0.bn.weight", "decode.head.out5.0.bn.bias", "decode.head.out5.0.bn.running_mean", "decode.head.out5.0.bn.running_var", "decode.head.out4.0.conv.weight", "decode.head.out4.0.bn.weight", "decode.head.out4.0.bn.bias", "decode.head.out4.0.bn.running_mean", "decode.head.out4.0.bn.running_var", "decode.head.out3.0.conv.weight", "decode.head.out3.0.bn.weight", "decode.head.out3.0.bn.bias", "decode.head.out3.0.bn.running_mean", "decode.head.out3.0.bn.running_var", "decode.head.out2.conv.weight", "decode.head.out2.bn.weight", "decode.head.out2.bn.bias", "decode.head.out2.bn.running_mean", "decode.head.out2.bn.running_var", "decode.binarize.0.weight", "decode.binarize.1.weight", "decode.binarize.1.bias", "decode.binarize.1.running_mean", "decode.binarize.1.running_var", "decode.binarize.3.weight", "decode.binarize.3.bias", "decode.binarize.4.weight", "decode.binarize.4.bias", "decode.binarize.4.running_mean", "decode.binarize.4.running_var", "decode.binarize.6.weight", "decode.binarize.6.bias", "decode.thresh.0.weight", "decode.thresh.1.weight", "decode.thresh.1.bias", "decode.thresh.1.running_mean", "decode.thresh.1.running_var", "decode.thresh.3.weight", "decode.thresh.3.bias", "decode.thresh.4.weight", "decode.thresh.4.bias", "decode.thresh.4.running_mean", "decode.thresh.4.running_var", "decode.thresh.6.weight", "decode.thresh.6.bias". 
	Unexpected key(s) in state_dict: "module.backbone.conv1.weight", "module.backbone.bn1.weight", "module.backbone.bn1.bias", "module.backbone.bn1.running_mean", "module.backbone.bn1.running_var", "module.backbone.bn1.num_batches_tracked", "module.backbone.layer1.0.conv1.weight", "module.backbone.layer1.0.bn1.weight", "module.backbone.layer1.0.bn1.bias", "module.backbone.layer1.0.bn1.running_mean", "module.backbone.layer1.0.bn1.running_var", "module.backbone.layer1.0.bn1.num_batches_tracked", "module.backbone.layer1.0.conv2.weight", "module.backbone.layer1.0.bn2.weight", "module.backbone.layer1.0.bn2.bias", "module.backbone.layer1.0.bn2.running_mean", "module.backbone.layer1.0.bn2.running_var", "module.backbone.layer1.0.bn2.num_batches_tracked", "module.backbone.layer1.1.conv1.weight", "module.backbone.layer1.1.bn1.weight", "module.backbone.layer1.1.bn1.bias", "module.backbone.layer1.1.bn1.running_mean", "module.backbone.layer1.1.bn1.running_var", "module.backbone.layer1.1.bn1.num_batches_tracked", "module.backbone.layer1.1.conv2.weight", "module.backbone.layer1.1.bn2.weight", "module.backbone.layer1.1.bn2.bias", "module.backbone.layer1.1.bn2.running_mean", "module.backbone.layer1.1.bn2.running_var", "module.backbone.layer1.1.bn2.num_batches_tracked", "module.backbone.layer2.0.conv1.weight", "module.backbone.layer2.0.bn1.weight", "module.backbone.layer2.0.bn1.bias", "module.backbone.layer2.0.bn1.running_mean", "module.backbone.layer2.0.bn1.running_var", "module.backbone.layer2.0.bn1.num_batches_tracked", "module.backbone.layer2.0.conv2.weight", "module.backbone.layer2.0.bn2.weight", "module.backbone.layer2.0.bn2.bias", "module.backbone.layer2.0.bn2.running_mean", "module.backbone.layer2.0.bn2.running_var", "module.backbone.layer2.0.bn2.num_batches_tracked", "module.backbone.layer2.0.downsample.0.weight", "module.backbone.layer2.0.downsample.1.weight", "module.backbone.layer2.0.downsample.1.bias", "module.backbone.layer2.0.downsample.1.running_mean", "module.backbone.layer2.0.downsample.1.running_var", "module.backbone.layer2.0.downsample.1.num_batches_tracked", "module.backbone.layer2.1.conv1.weight", "module.backbone.layer2.1.bn1.weight", "module.backbone.layer2.1.bn1.bias", "module.backbone.layer2.1.bn1.running_mean", "module.backbone.layer2.1.bn1.running_var", "module.backbone.layer2.1.bn1.num_batches_tracked", "module.backbone.layer2.1.conv2.weight", "module.backbone.layer2.1.bn2.weight", "module.backbone.layer2.1.bn2.bias", "module.backbone.layer2.1.bn2.running_mean", "module.backbone.layer2.1.bn2.running_var", "module.backbone.layer2.1.bn2.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.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.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.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.decode.head.in5.conv.weight", "module.decode.head.in5.bn.weight", "module.decode.head.in5.bn.bias", "module.decode.head.in5.bn.running_mean", "module.decode.head.in5.bn.running_var", "module.decode.head.in5.bn.num_batches_tracked", "module.decode.head.in4.conv.weight", "module.decode.head.in4.bn.weight", "module.decode.head.in4.bn.bias", "module.decode.head.in4.bn.running_mean", "module.decode.head.in4.bn.running_var", "module.decode.head.in4.bn.num_batches_tracked", "module.decode.head.in3.conv.weight", "module.decode.head.in3.bn.weight", "module.decode.head.in3.bn.bias", "module.decode.head.in3.bn.running_mean", "module.decode.head.in3.bn.running_var", "module.decode.head.in3.bn.num_batches_tracked", "module.decode.head.in2.conv.weight", "module.decode.head.in2.bn.weight", "module.decode.head.in2.bn.bias", "module.decode.head.in2.bn.running_mean", "module.decode.head.in2.bn.running_var", "module.decode.head.in2.bn.num_batches_tracked", "module.decode.head.out5.0.conv.weight", "module.decode.head.out5.0.bn.weight", "module.decode.head.out5.0.bn.bias", "module.decode.head.out5.0.bn.running_mean", "module.decode.head.out5.0.bn.running_var", "module.decode.head.out5.0.bn.num_batches_tracked", "module.decode.head.out4.0.conv.weight", "module.decode.head.out4.0.bn.weight", "module.decode.head.out4.0.bn.bias", "module.decode.head.out4.0.bn.running_mean", "module.decode.head.out4.0.bn.running_var", "module.decode.head.out4.0.bn.num_batches_tracked", "module.decode.head.out3.0.conv.weight", "module.decode.head.out3.0.bn.weight", "module.decode.head.out3.0.bn.bias", "module.decode.head.out3.0.bn.running_mean", "module.decode.head.out3.0.bn.running_var", "module.decode.head.out3.0.bn.num_batches_tracked", "module.decode.head.out2.conv.weight", "module.decode.head.out2.bn.weight", "module.decode.head.out2.bn.bias", "module.decode.head.out2.bn.running_mean", "module.decode.head.out2.bn.running_var", "module.decode.head.out2.bn.num_batches_tracked", "module.decode.binarize.0.weight", "module.decode.binarize.1.weight", "module.decode.binarize.1.bias", "module.decode.binarize.1.running_mean", "module.decode.binarize.1.running_var", "module.decode.binarize.1.num_batches_tracked", "module.decode.binarize.3.weight", "module.decode.binarize.3.bias", "module.decode.binarize.4.weight", "module.decode.binarize.4.bias", "module.decode.binarize.4.running_mean", "module.decode.binarize.4.running_var", "module.decode.binarize.4.num_batches_tracked", "module.decode.binarize.6.weight", "module.decode.binarize.6.bias", "module.decode.thresh.0.weight", "module.decode.thresh.1.weight", "module.decode.thresh.1.bias", "module.decode.thresh.1.running_mean", "module.decode.thresh.1.running_var", "module.decode.thresh.1.num_batches_tracked", "module.decode.thresh.3.weight", "module.decode.thresh.3.bias", "module.decode.thresh.4.weight", "module.decode.thresh.4.bias", "module.decode.thresh.4.running_mean", "module.decode.thresh.4.running_var", "module.decode.thresh.4.num_batches_tracked", "module.decode.thresh.6.weight", "module.decode.thresh.6.bias". 

Process finished with exit code 1

the-star-sea avatar Jun 29 '22 12:06 the-star-sea

when I run train_finetune,there is an error

Resuming from checkpoint.
Traceback (most recent call last):
  File "/home/DISCOVER_summer2022/zhangt/dblite/pruned/train_fintune.py", line 212, in <module>
    train_net(config)
  File "/home/DISCOVER_summer2022/zhangt/dblite/pruned/train_fintune.py", line 90, in train_net
    model.load_state_dict(checkpoint['state_dict'])
  File "/home/DISCOVER_summer2022/zhangt/.conda/envs/dbnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1625, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DBNet:
	Missing key(s) in state_dict: "backbone.conv1.weight", "backbone.bn1.weight", "backbone.bn1.bias", "backbone.bn1.running_mean", "backbone.bn1.running_var", "backbone.layer1.0.conv1.weight", "backbone.layer1.0.bn1.weight", "backbone.layer1.0.bn1.bias", "backbone.layer1.0.bn1.running_mean", "backbone.layer1.0.bn1.running_var", "backbone.layer1.0.conv2.weight", "backbone.layer1.0.bn2.weight", "backbone.layer1.0.bn2.bias", "backbone.layer1.0.bn2.running_mean", "backbone.layer1.0.bn2.running_var", "backbone.layer1.1.conv1.weight", "backbone.layer1.1.bn1.weight", "backbone.layer1.1.bn1.bias", "backbone.layer1.1.bn1.running_mean", "backbone.layer1.1.bn1.running_var", "backbone.layer1.1.conv2.weight", "backbone.layer1.1.bn2.weight", "backbone.layer1.1.bn2.bias", "backbone.layer1.1.bn2.running_mean", "backbone.layer1.1.bn2.running_var", "backbone.layer2.0.conv1.weight", "backbone.layer2.0.bn1.weight", "backbone.layer2.0.bn1.bias", "backbone.layer2.0.bn1.running_mean", "backbone.layer2.0.bn1.running_var", "backbone.layer2.0.conv2.weight", "backbone.layer2.0.bn2.weight", "backbone.layer2.0.bn2.bias", "backbone.layer2.0.bn2.running_mean", "backbone.layer2.0.bn2.running_var", "backbone.layer2.0.downsample.0.weight", "backbone.layer2.0.downsample.1.weight", "backbone.layer2.0.downsample.1.bias", "backbone.layer2.0.downsample.1.running_mean", "backbone.layer2.0.downsample.1.running_var", "backbone.layer2.1.conv1.weight", "backbone.layer2.1.bn1.weight", "backbone.layer2.1.bn1.bias", "backbone.layer2.1.bn1.running_mean", "backbone.layer2.1.bn1.running_var", "backbone.layer2.1.conv2.weight", "backbone.layer2.1.bn2.weight", "backbone.layer2.1.bn2.bias", "backbone.layer2.1.bn2.running_mean", "backbone.layer2.1.bn2.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.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.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.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", "decode.head.in5.conv.weight", "decode.head.in5.bn.weight", "decode.head.in5.bn.bias", "decode.head.in5.bn.running_mean", "decode.head.in5.bn.running_var", "decode.head.in4.conv.weight", "decode.head.in4.bn.weight", "decode.head.in4.bn.bias", "decode.head.in4.bn.running_mean", "decode.head.in4.bn.running_var", "decode.head.in3.conv.weight", "decode.head.in3.bn.weight", "decode.head.in3.bn.bias", "decode.head.in3.bn.running_mean", "decode.head.in3.bn.running_var", "decode.head.in2.conv.weight", "decode.head.in2.bn.weight", "decode.head.in2.bn.bias", "decode.head.in2.bn.running_mean", "decode.head.in2.bn.running_var", "decode.head.out5.0.conv.weight", "decode.head.out5.0.bn.weight", "decode.head.out5.0.bn.bias", "decode.head.out5.0.bn.running_mean", "decode.head.out5.0.bn.running_var", "decode.head.out4.0.conv.weight", "decode.head.out4.0.bn.weight", "decode.head.out4.0.bn.bias", "decode.head.out4.0.bn.running_mean", "decode.head.out4.0.bn.running_var", "decode.head.out3.0.conv.weight", "decode.head.out3.0.bn.weight", "decode.head.out3.0.bn.bias", "decode.head.out3.0.bn.running_mean", "decode.head.out3.0.bn.running_var", "decode.head.out2.conv.weight", "decode.head.out2.bn.weight", "decode.head.out2.bn.bias", "decode.head.out2.bn.running_mean", "decode.head.out2.bn.running_var", "decode.binarize.0.weight", "decode.binarize.1.weight", "decode.binarize.1.bias", "decode.binarize.1.running_mean", "decode.binarize.1.running_var", "decode.binarize.3.weight", "decode.binarize.3.bias", "decode.binarize.4.weight", "decode.binarize.4.bias", "decode.binarize.4.running_mean", "decode.binarize.4.running_var", "decode.binarize.6.weight", "decode.binarize.6.bias", "decode.thresh.0.weight", "decode.thresh.1.weight", "decode.thresh.1.bias", "decode.thresh.1.running_mean", "decode.thresh.1.running_var", "decode.thresh.3.weight", "decode.thresh.3.bias", "decode.thresh.4.weight", "decode.thresh.4.bias", "decode.thresh.4.running_mean", "decode.thresh.4.running_var", "decode.thresh.6.weight", "decode.thresh.6.bias". 
	Unexpected key(s) in state_dict: "module.backbone.conv1.weight", "module.backbone.bn1.weight", "module.backbone.bn1.bias", "module.backbone.bn1.running_mean", "module.backbone.bn1.running_var", "module.backbone.bn1.num_batches_tracked", "module.backbone.layer1.0.conv1.weight", "module.backbone.layer1.0.bn1.weight", "module.backbone.layer1.0.bn1.bias", "module.backbone.layer1.0.bn1.running_mean", "module.backbone.layer1.0.bn1.running_var", "module.backbone.layer1.0.bn1.num_batches_tracked", "module.backbone.layer1.0.conv2.weight", "module.backbone.layer1.0.bn2.weight", "module.backbone.layer1.0.bn2.bias", "module.backbone.layer1.0.bn2.running_mean", "module.backbone.layer1.0.bn2.running_var", "module.backbone.layer1.0.bn2.num_batches_tracked", "module.backbone.layer1.1.conv1.weight", "module.backbone.layer1.1.bn1.weight", "module.backbone.layer1.1.bn1.bias", "module.backbone.layer1.1.bn1.running_mean", "module.backbone.layer1.1.bn1.running_var", "module.backbone.layer1.1.bn1.num_batches_tracked", "module.backbone.layer1.1.conv2.weight", "module.backbone.layer1.1.bn2.weight", "module.backbone.layer1.1.bn2.bias", "module.backbone.layer1.1.bn2.running_mean", "module.backbone.layer1.1.bn2.running_var", "module.backbone.layer1.1.bn2.num_batches_tracked", "module.backbone.layer2.0.conv1.weight", "module.backbone.layer2.0.bn1.weight", "module.backbone.layer2.0.bn1.bias", "module.backbone.layer2.0.bn1.running_mean", "module.backbone.layer2.0.bn1.running_var", "module.backbone.layer2.0.bn1.num_batches_tracked", "module.backbone.layer2.0.conv2.weight", "module.backbone.layer2.0.bn2.weight", "module.backbone.layer2.0.bn2.bias", "module.backbone.layer2.0.bn2.running_mean", "module.backbone.layer2.0.bn2.running_var", "module.backbone.layer2.0.bn2.num_batches_tracked", "module.backbone.layer2.0.downsample.0.weight", "module.backbone.layer2.0.downsample.1.weight", "module.backbone.layer2.0.downsample.1.bias", "module.backbone.layer2.0.downsample.1.running_mean", "module.backbone.layer2.0.downsample.1.running_var", "module.backbone.layer2.0.downsample.1.num_batches_tracked", "module.backbone.layer2.1.conv1.weight", "module.backbone.layer2.1.bn1.weight", "module.backbone.layer2.1.bn1.bias", "module.backbone.layer2.1.bn1.running_mean", "module.backbone.layer2.1.bn1.running_var", "module.backbone.layer2.1.bn1.num_batches_tracked", "module.backbone.layer2.1.conv2.weight", "module.backbone.layer2.1.bn2.weight", "module.backbone.layer2.1.bn2.bias", "module.backbone.layer2.1.bn2.running_mean", "module.backbone.layer2.1.bn2.running_var", "module.backbone.layer2.1.bn2.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.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.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.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.decode.head.in5.conv.weight", "module.decode.head.in5.bn.weight", "module.decode.head.in5.bn.bias", "module.decode.head.in5.bn.running_mean", "module.decode.head.in5.bn.running_var", "module.decode.head.in5.bn.num_batches_tracked", "module.decode.head.in4.conv.weight", "module.decode.head.in4.bn.weight", "module.decode.head.in4.bn.bias", "module.decode.head.in4.bn.running_mean", "module.decode.head.in4.bn.running_var", "module.decode.head.in4.bn.num_batches_tracked", "module.decode.head.in3.conv.weight", "module.decode.head.in3.bn.weight", "module.decode.head.in3.bn.bias", "module.decode.head.in3.bn.running_mean", "module.decode.head.in3.bn.running_var", "module.decode.head.in3.bn.num_batches_tracked", "module.decode.head.in2.conv.weight", "module.decode.head.in2.bn.weight", "module.decode.head.in2.bn.bias", "module.decode.head.in2.bn.running_mean", "module.decode.head.in2.bn.running_var", "module.decode.head.in2.bn.num_batches_tracked", "module.decode.head.out5.0.conv.weight", "module.decode.head.out5.0.bn.weight", "module.decode.head.out5.0.bn.bias", "module.decode.head.out5.0.bn.running_mean", "module.decode.head.out5.0.bn.running_var", "module.decode.head.out5.0.bn.num_batches_tracked", "module.decode.head.out4.0.conv.weight", "module.decode.head.out4.0.bn.weight", "module.decode.head.out4.0.bn.bias", "module.decode.head.out4.0.bn.running_mean", "module.decode.head.out4.0.bn.running_var", "module.decode.head.out4.0.bn.num_batches_tracked", "module.decode.head.out3.0.conv.weight", "module.decode.head.out3.0.bn.weight", "module.decode.head.out3.0.bn.bias", "module.decode.head.out3.0.bn.running_mean", "module.decode.head.out3.0.bn.running_var", "module.decode.head.out3.0.bn.num_batches_tracked", "module.decode.head.out2.conv.weight", "module.decode.head.out2.bn.weight", "module.decode.head.out2.bn.bias", "module.decode.head.out2.bn.running_mean", "module.decode.head.out2.bn.running_var", "module.decode.head.out2.bn.num_batches_tracked", "module.decode.binarize.0.weight", "module.decode.binarize.1.weight", "module.decode.binarize.1.bias", "module.decode.binarize.1.running_mean", "module.decode.binarize.1.running_var", "module.decode.binarize.1.num_batches_tracked", "module.decode.binarize.3.weight", "module.decode.binarize.3.bias", "module.decode.binarize.4.weight", "module.decode.binarize.4.bias", "module.decode.binarize.4.running_mean", "module.decode.binarize.4.running_var", "module.decode.binarize.4.num_batches_tracked", "module.decode.binarize.6.weight", "module.decode.binarize.6.bias", "module.decode.thresh.0.weight", "module.decode.thresh.1.weight", "module.decode.thresh.1.bias", "module.decode.thresh.1.running_mean", "module.decode.thresh.1.running_var", "module.decode.thresh.1.num_batches_tracked", "module.decode.thresh.3.weight", "module.decode.thresh.3.bias", "module.decode.thresh.4.weight", "module.decode.thresh.4.bias", "module.decode.thresh.4.running_mean", "module.decode.thresh.4.running_var", "module.decode.thresh.4.num_batches_tracked", "module.decode.thresh.6.weight", "module.decode.thresh.6.bias". 

这是因为你在训练的时候开启了并行训练 在保存参数事自动加上了module.

fxwfzsxyq avatar Jul 01 '22 05:07 fxwfzsxyq