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的模型，所以不匹配

---

解决方案：