OpenLearning4DeepRecsys icon indicating copy to clipboard operation
OpenLearning4DeepRecsys copied to clipboard

GPU usage

Open attardi opened this issue 8 years ago • 1 comments

I have tried to run the code on a server with 4 Nvidia GPUS and the performance drops 6 times with respect to using a CPU on a notebook. Is there a way to exploit GPUs better?

attardi avatar Oct 31 '17 09:10 attardi

hi attardi, I have been working on some urgent projects recently and hopely will investigate the problem a.s.a.p. For a short try, could you please test replacing the l2 regularization (lines 339 to 350) with tf.nn.l2_loss() ? In our experience, it help boosting the program.

Leavingseason avatar Nov 08 '17 09:11 Leavingseason