GPU question

[sersoage]

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

[Shivanandroy]

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:

KeyPhraseTransformer/keyphrasetransformer/keyphrasetransformer.py

Line 72 in 80e2c8d

def predict(self, doc: str):

[kunalgoyal9]

#11