search

snipsco / ntm-lasagne

Neural Turing Machines library in Theano with Lasagne
https://medium.com/snips-ai/ntm-lasagne-a-library-for-neural-turing-machines-in-lasagne-2cdce6837315#.63t84s5r5
MIT License
300 stars 51 forks source link

The examples run more slowly on GPU than on CPU #32

Open andreasrobinson opened 8 years ago

andreasrobinson commented 8 years ago

Is this the expected behavior? Btw, I tested w/ a gtx-1070 gpu versus an i7-6700 CPU.

tristandeleu commented 8 years ago

I only had the opportunity to test it on GPU with an AWS g2.2xlarge/g2.8xlarge instances, and indeed it was as fast as running it on CPU. Unfortunately I don't have access to any dedicated machine with a GPU at the moment, so I can't run any further tests on that.