YOLOv5-Multibackbone-Compression icon indicating copy to clipboard operation
YOLOv5-Multibackbone-Compression copied to clipboard

运行pruneSlim.py时报错IndexError: list index out of range

Open dtde opened this issue 3 years ago • 0 comments

image 请问怎么解决?Traceback (most recent call last): File "pruneEagleEye.py", line 127, in rand_prune_and_eval(model,ignore_idx,opt) File "pruneEagleEye.py", line 59, in rand_prune_and_eval pruned_yaml = update_yaml(pruned_yaml, model, ignore_conv_idx, maskconvdict, opt) File "/hy-nas/yolov5-multibackbone/utils/prune_utils.py", line 263, in update_yaml update_yaml_loop(pruned_yaml,name,maskdict) File "/hy-nas/yolov5-multibackbone/utils/prune_utils.py", line 247, in update_yaml_loop args[-3][1] = 0.5 * maskconvdict[named_m_cv2_conv].sum().item() / (c2*0.5) IndexError: list index out of range

dtde avatar Jun 22 '22 08:06 dtde