deepspeech2 modify for kunlun

paddlespeech/s2t/exps/deepspeech2/bin/export.py

@@ -41,6 +41,7 @@ if __name__ == "__main__":
         '--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))

paddlespeech/s2t/exps/deepspeech2/bin/test.py

@@ -41,6 +41,7 @@ if __name__ == "__main__":
         '--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())

paddlespeech/s2t/exps/deepspeech2/bin/test_export.py

@@ -44,6 +44,7 @@ if __name__ == "__main__":
         '--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")

paddlespeech/s2t/exps/deepspeech2/bin/train.py

@@ -37,6 +37,7 @@ if __name__ == "__main__":
         '--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))