We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
TransformerBasedClassification and SetFitClassification should select a GPU if available and no device was set (device=None which is the default).
device=None
Other libraries that small-text uses, such as transformers, sentence-transformers, or setfit check if a GPU device is available and use it by default.
We should adapt the same behavior, this makes it easier for new users and CPU-only usage can still be forced by setting the device parameter.
Thanks to @eisioriginal for reporting
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Feature description
TransformerBasedClassification and SetFitClassification should select a GPU if available and no device was set (
device=None
which is the default).Motivation
Other libraries that small-text uses, such as transformers, sentence-transformers, or setfit check if a GPU device is available and use it by default.
We should adapt the same behavior, this makes it easier for new users and CPU-only usage can still be forced by setting the device parameter.
Additional comments
Thanks to @eisioriginal for reporting
The text was updated successfully, but these errors were encountered: