ptrendx commented on issue #7813: Performance doesn't improve (scalability issue) with # GPUs with running train_imagenet.py URL: https://github.com/apache/incubator-mxnet/issues/7813#issuecomment-328254930 You are CPU limited - Alexnet is a very small network and Titans are fast. How many cores do you have on your CPU? If more than 4 then try to set `--data-nthreads N` where N is number of threads to do IO. By default MXNet uses only 4 cores to do IO. You may also limit augmentations you perform (by default train_imagenet script performs a lot of augmentations). I am in the process of improving image IO (#7152), that should also help. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
With regards, Apache Git Services
