should I use GPU to train it?

tristandeleu / ntm-one-shot

One-shot Learning with Memory-Augmented Neural Networks

MIT License

421 stars 94 forks source link

should I use GPU to train it? #12

Closed BoyuanJiang closed 7 years ago

BoyuanJiang commented 7 years ago

If other specifications of computer are same(like cpu,ram),train it use GPU will faster than no GPU?If so,how many times faster?

tristandeleu commented 7 years ago

I can't say for sure about this model in particular, but I didn't get any significant speedup with a GPU on a vanilla NTM. If you end up using a GPU, please share your experience (if training was any faster).

BoyuanJiang commented 7 years ago

I tried to use gpu but even slower than cpu.