Update device usage.

3 years ago · 1707244472
parent 000294132c
commit 1707244472
2 changed files with 2 additions and 2 deletions
--- 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: