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

原创

已于 2022-04-13 15:17:28 修改 · 2.2w 阅读

108

CC 4.0 BY-SA版权

文章标签：

于 2022-04-01 23:30:25 首次发布

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

最近在学习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的模型，所以不匹配

200万优质内容无限畅学