Comments (1)
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)
- 稀疏训练之后,Binary Acc和Binary IoU都增加了几个点,正常吗? HOT 4
- IndexError: list index out of range HOT 15
- Errors in loading state_dict for DBNet HOT 5
- 关于DBhead和FPNhead性能问题? HOT 3
- 有没有libtorch实现的dbnet推理 HOT 1
- 关于多边形数据的验证问题 HOT 2
- cd models/dcn/ sh make.sh这个必须编译吗? HOT 1
- 跑几个epoch后,评估指标Binary Acc固定在0.5,正常吗? HOT 1
- LOSS在不断下降,但是一到验证阶段Recall,p,f1,全部为0 是什么原因? HOT 4
- pure is wrong HOT 1
- DB模型收敛的慢 HOT 2
- 能增加DB检测后将对应的目标区域裁剪的代码吗
- 用官方放出的权重测试也得不到正确的结果 HOT 2
- 您好,自定义数据集该如何制作呢? HOT 1
- model size(M) HOT 1
- 加载预训练模型 HOT 2
- 训练数据集 HOT 1
- recall HOT 5
- DB.pth.tar哪里可以下载? HOT 3
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from dbnet-lite.pytorch.