KeyPhraseTransformer icon indicating copy to clipboard operation
KeyPhraseTransformer copied to clipboard
verified publisher icon Shivanandroy

GPU question

Open sersoage opened this issue 3 years ago • 2 comments

Is it possible to run on GPU, if yes how? Thanks!

sersoage avatar Sep 19 '22 18:09 sersoage

Hey @sersoage : In the current version it only runs on CPU. I will add a flag to run GPU or CPU in the next release.

For now, you can fork it and then make changes in predict method to run it on GPU: https://github.com/Shivanandroy/KeyPhraseTransformer/blob/80e2c8d7c4869d1e120f6e8004630812443e766b/keyphrasetransformer/keyphrasetransformer.py#L72

Shivanandroy avatar Sep 20 '22 08:09 Shivanandroy

https://github.com/Shivanandroy/KeyPhraseTransformer/pull/11

kunalgoyal9 avatar Jun 23 '24 18:06 kunalgoyal9