Skip to content
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

[device args support] Replace gpu args with device args in train.py/recognize.py, etc. #2513

Open
MengqingCao opened this issue May 6, 2024 · 0 comments
Assignees

Comments

@MengqingCao
Copy link
Contributor

MengqingCao commented May 6, 2024

Is your feature request related to a problem? Please describe.
I'm running wenet on Ascend NPU device recently. And I find that the scripts, e.g., train.py, recognize.py, ./cli/modle.py, only support GPU device via --gpu argument.

Describe the solution you'd like
Specifying the accelerator to be used through --device argument will make it more convenient for people using wenet with different devices. There are a lot of devices (gpu, mps, xpu, etc.) supported in PyTorch, including Ascend NPU. If the replacement brings too many changes, adding a --device argumentcould also solve the problem, although I think the way of replacing is more reasonable.

Please let me know if there is any suggestions or concerns, thx!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants