Gumpest / YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
989 stars 201 forks source link

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

Open dtde opened 2 years ago

dtde commented 2 years ago

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() / (c20.5) IndexError: list index out of range