Update device usage.

17072444 · KP · 00029413 · 17072444 · 17072444
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

paddlespeech/cli/README.md paddlespeech/cli/README.md +1 -1

paddlespeech/cli/asr/infer.py paddlespeech/cli/asr/infer.py +1 -1

未找到文件。
--- a/paddlespeech/cli/README.md
+++ b/paddlespeech/cli/README.md
@@ -6,4 +6,4 @@
 `paddlespeech help`

 ## ASR
- `paddlespeech asr --input ./test_audio.wav --device gpu`
+ `paddlespeech asr --input ./test_audio.wav`
--- a/paddlespeech/cli/asr/infer.py
+++ b/paddlespeech/cli/asr/infer.py
@@ -87,7 +87,7 @@ class ASRExecutor(BaseExecutor):
        self.parser.add_argument(
            '--device',
            type=str,
-            default='cpu',
+            default=paddle.get_device(),
            help='Choose device to execute model inference.')

    def _get_pretrained_path(self, tag: str) -> os.PathLike: