RuntimeError: shape '[1024, 512, 3, 3]' is invalid for input of size 4554552

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏

Apache License 2.0

1.5k stars 446 forks source link

Open futureflsl opened 3 years ago

futureflsl commented 3 years ago

yolov4是直接用darknet yolov4训练出来的，根据lissue，我换成作者项目里面的yolov4.cfg还是报一样错误，怎么解决？

curryJ commented 2 years ago

yolov4是直接用darknet yolov4训练出来的，根据lissue，我换成作者项目里面的yolov4.cfg还是报一样错误，怎么解决？

请问解决了吗？权重文件载入shape不对