search

Syencil / mobile-yolov5-pruning-distillation

mobilev2-yolov5s剪枝、蒸馏，支持ncnn，tensorRT部署。ultra-light but better performence！
MIT License
821 stars 163 forks source link

RuntimeError: shape '[1, 3, 85, 80, 64]' is invalid for input of size 655360 #41

Open ahhhxxx opened 3 years ago

ahhhxxx commented 3 years ago

大佬您好，我运行detect.py报错 x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() RuntimeError: shape '[1, 3, 85, 80, 64]' is invalid for input of size 655360 运行train.py拿来训练也报错，是不是这不能拿来训练或者测试呢？

Syencil commented 3 years ago

1x35x85x80x64=1305600!=655360

yyyooohao commented 3 years ago

大佬您好，我运行detect.py报错 x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous() RuntimeError: shape '[1, 3, 85, 80, 64]' is invalid for input of size 655360 运行train.py拿来训练也报错，是不是这不能拿来训练或者测试呢？

你这个问题是怎么解决的呢，

yyyooohao commented 3 years ago

1x35x85x80x64=1305600!=655360

这咋改呢