MaartenGr / PolyFuzz

Fuzzy string matching, grouping, and evaluation.
https://maartengr.github.io/PolyFuzz/
MIT License
733 stars 67 forks source link

Support cuda #15

Closed chengren closed 3 years ago

chengren commented 3 years ago

Does the current package support GPU or are there any ways to call cuda?

MaartenGr commented 3 years ago

That depends on the underlying model. If you are using Flair, then the model is automatically executed on the GPU if it is available. You can also specify the device as such:

import flair, torch
flair.device = torch.device('cpu') 

All other models, such as TF-IDF and Levenshtein do not support GPU-based processing.