search

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

keyerror #150

Open lccui opened 3 years ago

lccui commented 3 years ago
微信图片
lccui commented 3 years ago

percent_limit = (sorted_bn==highest_thre).nonzero(as_tuple=False).item()/len(bn_weights) ValueError: only one element tensors can be converted to Python scalars

lccui commented 3 years ago

prune_util.py mask = CBLidx2mask[route_in_idxs[0]] KeyError: 108