search

stanfordnlp / treelstm

Tree-structured Long Short-Term Memory networks (http://arxiv.org/abs/1503.00075)
GNU General Public License v2.0
875 stars 236 forks source link

Can this be configured to run on GPU? #9

Closed hardikmeisheri closed 7 years ago

hardikmeisheri commented 7 years ago

I have tried to look up on net but we need to specify each tensor that is being used in this to be used as GPU tensor, Is there any other way to run entire code on GPU?

kaishengtai commented 7 years ago

While GPU execution is not currently supported, it should be possible to implement it by adding the appropriate :cuda() calls.

For an out-of-the-box solution, try checking out the Tensorflow implementation: https://github.com/tensorflow/fold/blob/master/tensorflow_fold/g3doc/sentiment.ipynb

hardikmeisheri commented 7 years ago

@kaishengtai Thanks for the link.