mxnet-SSH icon indicating copy to clipboard operation
mxnet-SSH copied to clipboard

training speed

Open bruinxiong opened this issue 7 years ago • 1 comments

I run training code, but the speed is quite slow. only 20 images/sec even I set batch_image =32 on 1 GPU or 4 GPU even larger. Could you give some comments ?

bruinxiong avatar Dec 05 '18 05:12 bruinxiong

CPU is the performance bottleneck. We may need a refined c++ data loader to boost the speed.

nttstar avatar Dec 05 '18 06:12 nttstar