-
Notifications
You must be signed in to change notification settings - Fork 2.5k
New issue
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
fix device setting to allow using accelerator cpu #8084
Conversation
Signed-off-by: Oren Amsalem <oren.a4@gmail.com>
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Makes sense, thanks for the fix ! @tbartley94 for final review
jenkins |
jenkins |
@titu1994 i know you pushed but just for record I'm not seeing any issues so review is approved |
Oh thanks for the update, I think I misremembered as this having your approval on slack somehow, turns out it was a different topic different pr. Sorry about that. |
do you @titu1994 or @tbartley94 have any idea about this one: |
Hmm, lemme take a look. I'd wager it's some tech debt between PTL versions. (fervor for wav2vec dipped before PTL 2.0 was released, so the code is likely working off a dated namespace.) |
* fix device setting to allow using accelerator cpu Signed-off-by: Oren Amsalem <oren.a4@gmail.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Oren Amsalem <oren.a4@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Signed-off-by: Piotr Żelasko <petezor@gmail.com>
* fix device setting to allow using accelerator cpu Signed-off-by: Oren Amsalem <oren.a4@gmail.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Oren Amsalem <oren.a4@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Signed-off-by: Sasha Meister <ameister@nvidia.com>
* fix device setting to allow using accelerator cpu Signed-off-by: Oren Amsalem <oren.a4@gmail.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Oren Amsalem <oren.a4@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
What does this PR do ?
It is currently impossible to run wav2vec with
trainer.accelerator=cpu
if the machine has a gpu.The padding_mask is already on the device set in
trainer.accelerator
but if the machine has a GPU the DEVICE variable will contain cuda and this error message will be received:PR Type:
Additional Information
@titu1994, @redoctopus, @jbalam-nv, or @okuchaiev