unable to prune yolov3 spp model. Need help

coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本，以适应不同的需求。

794 stars 215 forks source link

Open joel5638 opened 4 years ago

joel5638 commented 4 years ago

When I'm running python3 normal_prune.py

I get this error

TypeError: 'MaxPool2d' object is not subscriptable

can someone help? please

ZhihangTang commented 4 years ago

Hello,I need to prune yolov3-spp model,and I got this error too.Did you solve it？