ethanhe42 / channel-pruning

Channel Pruning for Accelerating Very Deep Neural Networks (ICCV'17)
https://arxiv.org/abs/1707.06168
MIT License
1.07k stars 310 forks source link

Excuse me, is there any version that support tensorflow ? #93

Closed LouiValley closed 5 years ago

LouiValley commented 6 years ago

Exceuse me, as I see, it implements under caffe. If I wanna prune some networks which is run under tensorflow implementation, is there any version that support tensorflow ?

young-fire commented 6 years ago

I want to do this , but need help.

ethanhe42 commented 5 years ago

You might be interested in pocketflow, which is implemented using tensorflow. They implemented both channel pruning and AMC. https://github.com/Tencent/PocketFlow#channel-pruning