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
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