Gumpest/YOLOv5-Multibackbone-Compression

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

dtde opened this issue · 0 comments

dtde commented

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