Open ricky1203 opened 7 years ago
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!
It looks like someone have implemented the gpu version here
@0bserver07 Looks great! Thanks!
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).
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