Giter VIP home page Giter VIP logo

Comments (1)

fxwfzsxyq avatar fxwfzsxyq commented on June 8, 2024

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.

from dbnet-lite.pytorch.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.