Hello author, since I have millions of documents, I want to use GPU to accelerate the computation of the KeyBERT algorithm, especially during the embedding phase. However, I don’t see a place to set the batch_size parameter. Are there any methods to speed up the computation? Thank you.
Hello author, since I have millions of documents, I want to use GPU to accelerate the computation of the KeyBERT algorithm, especially during the embedding phase. However, I don’t see a place to set the batch_size parameter. Are there any methods to speed up the computation? Thank you.