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

使用使用slim_prune.py时出现了下面问题 #161

Open zhaoleo1111 opened 3 years ago

zhaoleo1111 commented 3 years ago

runtimeerror:Given Groups=1, weight ofsize 1024 160 1 1,except input [16, 1, 1, 1] to have 1024 channels, but got 1 channels instead