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
https://pytorch.org/docs/master/tensor_attributes.html#torch-device Now this function only return cpu or cuda. I failed to run some transformer based models on my MacOS due to this limitation. I fixed it by calling model.to('mps').
cpu
cuda
The text was updated successfully, but these errors were encountered:
Please check my quick fix: #137
Sorry, something went wrong.
No branches or pull requests
https://pytorch.org/docs/master/tensor_attributes.html#torch-device
Now this function only return
cpu
orcuda
. I failed to run some transformer based models on my MacOS due to this limitation.I fixed it by calling model.to('mps').
The text was updated successfully, but these errors were encountered: