Partial GPU support in images

cpllab / lm-zoo

Easy black-box access to state-of-the-art language models

https://cpllab.github.io/lm-zoo/

MIT License

14 stars 6 forks source link

Partial GPU support in images #49

Closed hans closed 4 years ago

hans commented 4 years ago

No support on API/frontend for this, but ordered-neurons and tinylstm now support reading an LMZOO_USE_GPU flag and setting Torch device accordingly.

hans commented 4 years ago

cf #46