search

he-y / soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks
https://arxiv.org/abs/1808.06866
376 stars 74 forks source link

get small model #24

Open chaos1992 opened 4 years ago

chaos1992 commented 4 years ago

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!

yushuinanrong commented 4 years ago

@chaos1992 Have you solved this problem?

chaos1992 commented 4 years ago

@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