From b0eaeccd670b53ac77acd044fa738dca92f20032 Mon Sep 17 00:00:00 2001
From: Zhangjingyu06 <zhangjingyu06@baidu.com>
Date: Tue, 24 May 2022 11:58:49 +0000
Subject: [PATCH] deepspeech2 modify for kunlun

---
 paddlespeech/s2t/exps/deepspeech2/bin/export.py      | 2 +-
 paddlespeech/s2t/exps/deepspeech2/bin/test.py        | 2 +-
 paddlespeech/s2t/exps/deepspeech2/bin/test_export.py | 2 +-
 paddlespeech/s2t/exps/deepspeech2/bin/train.py       | 2 +-
 paddlespeech/s2t/training/trainer.py                 | 3 +++
 5 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/paddlespeech/s2t/exps/deepspeech2/bin/export.py b/paddlespeech/s2t/exps/deepspeech2/bin/export.py
index ae43bf82c..f1b84969d 100644
--- a/paddlespeech/s2t/exps/deepspeech2/bin/export.py
+++ b/paddlespeech/s2t/exps/deepspeech2/bin/export.py
@@ -40,7 +40,7 @@ if __name__ == "__main__":
     parser.add_argument(
         '--nxpu',
         type=int,
-        default=1,
+        default=0,
         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
index f29f50832..c91be411a 100644
--- a/paddlespeech/s2t/exps/deepspeech2/bin/test.py
+++ b/paddlespeech/s2t/exps/deepspeech2/bin/test.py
@@ -40,7 +40,7 @@ if __name__ == "__main__":
     parser.add_argument(
         '--nxpu',
         type=int,
-        default=1,
+        default=0,
         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
index c136ddf29..923c3db49 100644
--- a/paddlespeech/s2t/exps/deepspeech2/bin/test_export.py
+++ b/paddlespeech/s2t/exps/deepspeech2/bin/test_export.py
@@ -43,7 +43,7 @@ if __name__ == "__main__":
     parser.add_argument(
         '--nxpu',
         type=int,
-        default=1,
+        default=0,
         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
index cb4867ef2..d80cd0cb2 100644
--- a/paddlespeech/s2t/exps/deepspeech2/bin/train.py
+++ b/paddlespeech/s2t/exps/deepspeech2/bin/train.py
@@ -36,7 +36,7 @@ if __name__ == "__main__":
     parser.add_argument(
         '--nxpu',
         type=int,
-        default=1,
+        default=0,
         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/training/trainer.py b/paddlespeech/s2t/training/trainer.py
index d30556ca1..a7eb9892d 100644
--- a/paddlespeech/s2t/training/trainer.py
+++ b/paddlespeech/s2t/training/trainer.py
@@ -119,6 +119,9 @@ class Trainer():
                 paddle.set_device('xpu')
         elif self.args.ngpu > 0:
             paddle.set_device("gpu")
+        else:
+            raise Exception("invalid device")
+
         if self.parallel:
             self.init_parallel()