how to use gpu to train the model

[ricky1203]

will you please add code to use GPU to train the model, i am not familiar with chainer or anyone can port the code to tensorflow thanks

[physicso]

Although I'm not familiar with Chainer too, I have found a reference of using GPUs in it: http://docs.chainer.org/en/stable/tutorial/gpu.html Hope that will help!

[0bserver07]

It looks like someone have implemented the gpu version here

[physicso]

@0bserver07 Looks great! Thanks!

[AdeelMufti]

I created an optimized implementation, extending yos1up's: https://github.com/AdeelMufti/DifferentiableNeuralComputer.

It's around 50x faster, and can run on GPU (which is only beneficial for large memory sizes).