search

coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本，以适应不同的需求。
793 stars 214 forks source link

对yolov3-spp进行剪枝的时候遇到以下问题 #90

Open ZhihangTang opened 4 years ago

ZhihangTang commented 4 years ago

Traceback (most recent call last): File "D:/python projects/YOLOv3-complete-pruning/regular_prune.py", line 233, in pruned_model = prune_model_keep_size(model, prune_idx, CBL_idx, CBLidx2mask) File "D:\python projects\YOLOv3-complete-pruning\utils\prune_utils.py", line 221, in prune_model_keep_size next_conv = pruned_model.module_list[next_idx][0] TypeError: 'MaxPool2d' object does not support indexing