YOLO模型train.py文件报错

问题

在训练YOLO模型后将自己权重用于预测时总是提示如下报错信息：
RuntimeError: Error(s) in loading state_dict for YoloBody:
size mismatch for last_layer0.6.weight: copying a param with shape torch.Size([24, 1024, 1, 1]) from checkpoint, the shape in current model is torch.Size([27, 1024, 1, 1]).
。。。。。。。。
此处省略

解决办法:

将程序中 config.py下的class数改为需要预测的类别数+1，比如需要预测猫与狗，则class=3，因为需要加上背景，否则在预测时会出现错误，导致白白训练。

YOLO模型train.py文件报错：size mismatch for last_layer0.6.weight: copying a param with shape torch.相关推荐

size mismatch for yolo_head2.1.bias: copying a param with shape torch.Size(【75】) from checkpoint...
凯哥英语视频今天一个朋友用YOLO4预测图片报错:size mismatch for yolo_head2.1.bias: copying a param with shape torch.Size ...
size mismatch for fc.weight: copying a param with shape torch.Size([1000, 2048]) from checkpoint, th
问题描述我想在我自己的项目更换其他的模型,下载的预训练模型出现了FC层不匹配的问题,找了好多人都写了这个点,今天总结一下: 首先我们遇到的问题如下: 他的意思是resnet50的fc层是1000分类 ...
size mismatch for roi_heads.box_predictor.cls_score.weight: copying a param with shape torch.Size([9
1. 报错 RuntimeError: Error(s) in loading state_dict for FasterRCNN: size mismatch for roi_heads.box_p ...
Python pytest main.py文件报错main.py: error: unrecognized arguments: --html=./test_report.html问题解决
ERROR: usage: main.py [options] [file_or_dir] [file_or_dir] [...] main.py: error: unrecognized argum ...
python导入py文件报错
python有大量的内置模块,可以方便地导入.同样python也可以导入其它人写的py程序,以利用别人写好的类和函数. 1.在缺省状态下,导入一个已编写好的py文件用import 跟上py文件名,有时 ...
python中import文件夹下面py文件,报错
在需要导入的那个文件夹里面新建一个 __init__.py文件,哪怕这个文件是空的文件也可以.
解决使用ICsharpCode解压缩时候报错Size MisMatch的错误
项目用到了这个组件,然后在解压文件时候报Size MisMatch错,解决方法:到https://github.com/icsharpcode/SharpZipLib/releases选择对应的源码下 ...
深度学习 cnn pytorch框架报错 size mismatch, m1: [ x ], m2: [ x ] Captcha.ImageCaptcha设置生成图片的大小尺寸
看总结直接拉最后这里很大程度是因为图片大小不一致,可以通过 resize命令来进行图片的归一化. 如果是通过captcha库里面的ImageCaptcha生成的验证码,百度上没有任何资料说可以改变 ...
strict=False 但还是size mismatch for []: copying a param with shape [] from checkpoint,the shape in cur
strict=False 但还是size mismatch for []: copying a param with shape [] from checkpoint,the shape in cur ...
【项目笔记】布局文件报错Suspicious size: this will make the view invisible, probably intended for layout_width
写着写着就懵逼了,一直以为布局文件没写错啊,horizontal就是竖直啊,原来布局文件报错,不仅仅需要从报错的地方解决问题,还需要从其他地方去分析. 很明显是方向orientation选错了,应该写 ...

YOLO模型train.py文件报错：size mismatch for last_layer0.6.weight: copying a param with shape torch.

YOLO模型train.py文件报错

问题

解决办法:

YOLO模型train.py文件报错：size mismatch for last_layer0.6.weight: copying a param with shape torch.相关推荐

最新文章

热门文章