RuntimeError: The expanded size of the tensor (1) must match the existing size (80) at non-singleton dimension 3. Target sizes: [1, 3, 1, 1, 2]. Tensor sizes: [3, 80, 80, 2]

[Liuruifeng29]

RuntimeError: The expanded size of the tensor (1) must match the existing size (80) at non-singleton dimension 3. Target sizes: [1, 3, 1, 1, 2]. Tensor sizes: [3, 80, 80, 2]

有遇到这个问题的吗，换了一个权重文件就报这个错误

[wujunyi1412]

RuntimeError: The expanded size of the tensor (1) must match the existing size (80) at non-singleton dimension 3. Target sizes: [1, 3, 1, 1, 2]. Tensor sizes: [3, 80, 80, 2]

有遇到这个问题的吗，换了一个权重文件就报这个错误

你模型的版本错了，试试6.1

[Liuruifeng29]

就是我用其他的权重文件没问题，换了一个就报这个错误了

[Cinnaio]

我也是这个报错 训练模型用的是yolov5 version5
但是换成了https://github.com/cnyvfang/labelGo-Yolov5AutoLabelImg/releases/tag/source 就好了

该项目已支持最新版本YOLOv5，如需要使用支持YOLOv5 version5的旧版本，可以在Release中找到旧版本代码。

这个是作者在readme里面写的