You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Could you please let me know if only changing the following line is enough to run the code in GPU. Environment.getInstance().setExecutionMode(EXECUTION_MODE.SEQ); to Environment.getInstance().setExecutionMode(EXECUTION_MODE.GPU);
#36
Open
sudiptap opened this issue
Mar 18, 2015
· 4 comments
I've tried GPU, CPU, and SEQ for MINST testLeNetSmall, testSigmoidHiddenBP and XorTest testCNNMLPBP and none of them give a speed-up for GPU. In fact, sometimes CPU is faster, but never GPU. Can anyone give an example of a network that would benefit from being run on the GPU?
I got the GPU run faster, but only by making huge networks that would take forever to complete for example I modified the testLenetSmall function to have this network:
Basically I added a 3rd convolutional net, bumped up the number of filters in in all covnets to 120 (from 20 and 50), quadrupled the neurons in the final hidden layer and added another hidden layer with 2048 neurons. The GPU enabled version runs about 2.4 times faster, but it's still dog slow taking something like 12 - 14 seconds per batch (the batch size is 1) so training the entire dataset of 60000 images would take 8.3 to 9.7 days. So like 10 days per epoch on the GPU. Meanwhile I built a comparable network in Lasagne/Theano and it takes around 420 seconds per epoch on the CPU (in a VM at that) which is about 2000 times faster.
I have the required environment set up. In fact by just changing that I am not able to cut down the running time of my application. Thank you.
The text was updated successfully, but these errors were encountered: