deepspeech2 modify for kunlun

PaddlePaddle · May 24, 2022 · acb19cf · acb19cf
1 parent b0eaecc
commit acb19cf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/export.py b/paddlespeech/s2t/exps/deepspeech2/bin/export.py
@@ -41,6 +41,7 @@ def main(config, args):
         '--nxpu',
         type=int,
         default=0,
+        choices=[0, 1],
         help="if nxpu == 0 and ngpu == 0, use cpu.")
     args = parser.parse_args()
     print("model_type:{}".format(args.model_type))

diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/test.py b/paddlespeech/s2t/exps/deepspeech2/bin/test.py
@@ -41,6 +41,7 @@ def main(config, args):
         '--nxpu',
         type=int,
         default=0,
+        choices=[0, 1],
         help="if nxpu == 0 and ngpu == 0, use cpu.")
     args = parser.parse_args()
     print_arguments(args, globals())

diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/test_export.py b/paddlespeech/s2t/exps/deepspeech2/bin/test_export.py
@@ -44,6 +44,7 @@ def main(config, args):
         '--nxpu',
         type=int,
         default=0,
+        choices=[0, 1],
         help="if nxpu == 0 and ngpu == 0, use cpu.")
     parser.add_argument(
         "--enable-auto-log", action="store_true", help="use auto log")

diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/train.py b/paddlespeech/s2t/exps/deepspeech2/bin/train.py
@@ -37,6 +37,7 @@ def main(config, args):
         '--nxpu',
         type=int,
         default=0,
+        choices=[0, 1],
         help="if nxpu == 0 and ngpu == 0, use cpu.")
     args = parser.parse_args()
     print("model_type:{}".format(args.model_type))