soft-filter-pruning
soft-filter-pruning copied to clipboard
get small model
Hi, I notice that your get_small_model.py just can be used for resnet. But I want to use this code for other network. Could you give me some advice? Look forward to your reply, thanks!
@chaos1992 Have you solved this problem?
@yushuinanrong not yet. I have used amc, MIT work, to prune network finally. I think amc is more easy for getting small model. Code Link: https://github.com/mit-han-lab/amc