search

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

GPU #110

Closed shiruipeng1985 closed 5 years ago

shiruipeng1985 commented 5 years ago

All are welcome to create issues, but please google the problem first, and make sure it has not already been reported.

What steps reproduce the bug?

What hardware and operating system/distribution are you running?

Operating system:
CUDA version:
CUDNN version:
openCV version:
BLAS:
Python version:

If the bug is a crash, provide the backtrace.

shiruipeng1985 commented 5 years ago

Did the resnet50-2x model just be used in GPU mode?

ethanhe42 commented 5 years ago

yes