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 likeSpecifying 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!
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!