wenwei202 / iss-rnns

Sparse Recurrent Neural Networks -- Pruning Connections and Hidden Sizes (TensorFlow)
Apache License 2.0
73 stars 21 forks source link

Run with gpu? #7

Closed YikSanChan closed 6 years ago

YikSanChan commented 6 years ago

Hi,

Is it possible to run the code with gpu support easily? Say, with some command line arguments added? If so, could you please point it out?

Thank you!

wenwei202 commented 6 years ago

By default it runs on GPU, and I don't understand your question

YikSanChan commented 6 years ago

Thank you.