Syencil / mobile-yolov5-pruning-distillation

mobilev2-yolov5s剪枝、蒸馏,支持ncnn,tensorRT部署。ultra-light but better performence!
MIT License
816 stars 164 forks source link

遇到了点问题 #44

Open Lihong1997 opened 3 years ago

Lihong1997 commented 3 years ago

RuntimeError: Given groups=1, weight of size [1, 255, 1, 1], expected input[1, 512, 8, 8] to have 255 channels, but got 512 channels instead 我运行那个mob-yolov5s.yaml,报错这个!