You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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!
The text was updated successfully, but these errors were encountered:
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!
The text was updated successfully, but these errors were encountered: