ethanhe42/channel-pruning

GPU

Closed this issue · 2 comments

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.

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

yes