search

bedapudi6788 / txt2txt

Extremely easy to use sequence to sequence library with attention, for text to text conversion tasks.
http://bpraneeth.com/projects
GNU General Public License v3.0
39 stars 14 forks source link

Support TPU #4

Open muhammetfaik opened 4 years ago

muhammetfaik commented 4 years ago

It's my issue:https://github.com/tensorflow/tensorflow/issues/37054

AASHISHAG commented 4 years ago

@bedapudi6788 : I am running txt2txt on a large corpus for punctuation correction. But the process is running on CPU's. On checking GPU usage (nvidia-smi), I can see that some memory is allocated to GPU's but their usage is 0%. Did I miss anything? Please guide.

bedapudi6788 commented 4 years ago

Check if tensorflow can use gpu.

AASHISHAG commented 4 years ago

@bedapudi6788 : Thank you for the response. I have now installed tensflow-gpu and can see 1 gpu being used. Any idea how to use all the GPU's?

bedapudi6788 commented 4 years ago

All this information is a trivial google search away. I am happy to help if there are any problems with txt2txt.

EDIT: https://datascience.stackexchange.com/questions/23895/multi-gpu-in-keras