Channel selection cost too much time

[BowieHsu]

scripts is "./scripts/run_local.sh nets/mobilenet_at_ilsvrc12_run.py -n=1 --learner channel", �using "ChannelPrunedLeaner" method ,I check the pruning learner source code，maybe the program stacked at line504 while loop or line515 while loop?

[jiaxiang-wu]

Hi @psyyz10 can you take a look at this issue?

[psyyz10]

@BowieHsu Hi, yes, for mobilenet v2, the channel selection is slow, and for other networks the speed is ok. We will improve that by a further GPU implementation, please wait some days.

[BowieHsu]

@psyyz10 many thanks, so if i rewrite all those numpy code using tensorflow op, the channel selection speed should be improved a lot, am i right?

[psyyz10]

@BowieHsu Yes of course, if you can do that, your contributions are very welcome.

[jiaxiang-wu]

In pull request #84, we provide a GPU-based implementation for channel pruning algorithms. Numpy code is now replaced by TensorFlow operations.