Pytorch中报错RuntimeError: The size of tensor a (60) must match the size of tensor b (56)

YOLOV5中报错:RuntimeError: The size of tensor a (60) must match the size of tensor b (56) at non-singleton dimension 3


YOLOV5

最近在学习YOLOV5的时候,刚开始遇到了如下的问题:

RuntimeError: The size of tensor a (60) must match the size of tensor b (56) at non-singLeton dimension 3

原因分析:

这可能是因为5.0的工程下载了个6.1的模型,所以不匹配


解决方案:

yolov5s.pt
[https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt]
下载下来替换掉工程文件夹里的yolov5s.pt文件
发现下面这个问题直接消失了!

RuntimeError: The size of tensor a (60) must match the size of tensor b (56) at non-singLeton dimension 3

顺利完成
在这里插入图片描述

如果对你有用麻烦点个赞,谢谢啦!文章来源地址https://uudwc.com/A/nPWpV

原文地址:https://blog.csdn.net/qq_43108889/article/details/123910494

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请联系站长进行投诉反馈,一经查实,立即删除!

上一篇 2023年09月12日 06:14
下一篇 2023年09月12日 06:15