Any suggestions on how can I extend this code for MobileNetV2? about pytorch-pruning HOT 9 OPEN

Did you find a solution to this?

from pytorch-pruning.

@aprams Yes I did, had to do multiple changes to make that work. I basically flattened the mobilenetv2 model into a flat model before feeding into this code (dealing with the residual links is the trickiest part in the flattening process). Secondly, I changed prune.py significantly to address batchnorm layers, and depthwise convs. I hope that helps.

from pytorch-pruning.

@nekulkarni Can you share me your code,please? I need to prune the mobilenetv2 for the school project but I am the beginner for python.

from pytorch-pruning.

@aprams Yes I did, had to do multiple changes to make that work. I basically flattened the mobilenetv2 model into a flat model before feeding into this code (dealing with the residual links is the trickiest part in the flattening process). Secondly, I changed prune.py significantly to address batchnorm layers, and depthwise convs. I hope that helps.

Thanks for sharing, could you please share more about how to deal with batchnorm layers? Thanks so much.

from pytorch-pruning.

@nekulkarni , I have pruned mobilenetv2 correctly, but the pruned model is difficult to train on imagenet, the accuracy is very low( I train with a 1080Ti, batchsize=96, lr=0.045, weight decay=0.00004, and decrease the lr 0.98 for each epoch, and i also try lr =0.1, lr = lr * (0.1 ** (epoch // 30)) ). can you tell me your data set and accuracy.
thank you very much!

from pytorch-pruning.

It seems that mobilenetv2 convergence slowly !

from pytorch-pruning.

@nekulkarni Can you share me your code about mobilenetv2 pruned ,please? I am a beginner. Thank you very much!

from pytorch-pruning.

@viviov Can you share me your code about mobilenetv2 pruned ,please? I am a beginner. Thank you very much!

from pytorch-pruning.

@nekulkarni Can you share the code with me also? I want to know how to deal with batch norm layers. Thank you very much.!!

from pytorch-pruning.