search

yos1up / DNC

Differentiable Neural Computers
129 stars 34 forks source link

how to use gpu to train the model #1

Open ricky1203 opened 7 years ago

ricky1203 commented 7 years ago

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 commented 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!

0bserver07 commented 7 years ago

It looks like someone have implemented the gpu version here

physicso commented 7 years ago

@0bserver07 Looks great! Thanks!

AdeelMufti commented 6 years ago

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).