如何解决如何解决Yolov5 train,py in yaml中的错误
我正在尝试在 Google colab GPU 免费版中运行 yolov5
!pip install PyYAML==5.3
Model Summary: 407 layers,8.84875e+07 parameters,8.84875e+07 gradients
Optimizer groups: 134 .bias,142 conv.weight,131 other
Traceback (most recent call last):
File "/content/yolov5/train.py",line 116,in train
ckpt['model'] = {k: v for k,v in ckpt['model'].float().state_dict().items()
File "/content/yolov5/train.py",line 117,in <dictcomp>
if model.state_dict()[k].shape == v.shape} # to FP32,filter
KeyError: 'model.18.conv.weight'
我在 google colab 中的命令是
!python /content/yolov5/train.py --img 640 --batch 4 --epochs 30 \
--data /content/yolov5/data/clothing.yaml
--cfg /content/yolov5/models/yolov5x.yaml \
--weights yolov5x.pt \
--name yolov5_clothing --cache
你能帮我解决这个问题吗?
谢谢
解决方法
yolo 团队要求我运行 requirements.txt 并重新下载 yolo 已解决此问题。
一切都很好,您可以在 https://github.com/ultralytics/yolov5/issues/2181
上查看更多详细信息谢谢
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。