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://www.toymoban.com/news/detail-789405.html
如果对你有用麻烦点个赞,谢谢啦!文章来源地址https://www.toymoban.com/news/detail-789405.html
到了这里,关于Pytorch中报错RuntimeError: The size of tensor a (60) must match the size of tensor b (56)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!