提交 17072444 编写于 作者: K KP

Update device usage.

上级 00029413
...@@ -6,4 +6,4 @@ ...@@ -6,4 +6,4 @@
`paddlespeech help` `paddlespeech help`
## ASR ## ASR
`paddlespeech asr --input ./test_audio.wav --device gpu` `paddlespeech asr --input ./test_audio.wav`
...@@ -87,7 +87,7 @@ class ASRExecutor(BaseExecutor): ...@@ -87,7 +87,7 @@ class ASRExecutor(BaseExecutor):
self.parser.add_argument( self.parser.add_argument(
'--device', '--device',
type=str, type=str,
default='cpu', default=paddle.get_device(),
help='Choose device to execute model inference.') help='Choose device to execute model inference.')
def _get_pretrained_path(self, tag: str) -> os.PathLike: def _get_pretrained_path(self, tag: str) -> os.PathLike:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册