search

dmmiller612 / bert-extractive-summarizer

Easy to use extractive text summarization with BERT
MIT License
1.4k stars 308 forks source link

can we implement it with GPU support? #36

Closed zrxbeijing closed 4 years ago

zrxbeijing commented 4 years ago

Hello, I like your project very much. One question: how can we run the code with gpu support?

dmmiller612 commented 4 years ago

Haven’t tried it yet. I probably would need to include the device as an argument

jackall commented 4 years ago

This project is just awesome! thanks, @dmmiller612, but it would be great to squeeze it to GPU, CPU takes forever to run.

dmmiller612 commented 4 years ago

So sorry this took so long. GPU support has just been added. I will be uploading the code to pypi in the next few hours. The current colab in the repo has an example with gpu support (If you have a cuda available, you shouldn't need to do anything on your part with the most recent code change).

zrxbeijing commented 4 years ago

So sorry this took so long. GPU support has just been added. I will be uploading the code to pypi in the next few hours. The current colab in the repo has an example with gpu support (If you have a cuda available, you shouldn't need to do anything on your part with the most recent code change).

Wow! That‘s awesome! Can’t wait to try it out!