提交 b0eaeccd 编写于 作者: Z Zhangjingyu06

deepspeech2 modify for kunlun

上级 1e91f7da
...@@ -40,7 +40,7 @@ if __name__ == "__main__": ...@@ -40,7 +40,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
'--nxpu', '--nxpu',
type=int, type=int,
default=1, default=0,
help="if nxpu == 0 and ngpu == 0, use cpu.") help="if nxpu == 0 and ngpu == 0, use cpu.")
args = parser.parse_args() args = parser.parse_args()
print("model_type:{}".format(args.model_type)) print("model_type:{}".format(args.model_type))
......
...@@ -40,7 +40,7 @@ if __name__ == "__main__": ...@@ -40,7 +40,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
'--nxpu', '--nxpu',
type=int, type=int,
default=1, default=0,
help="if nxpu == 0 and ngpu == 0, use cpu.") help="if nxpu == 0 and ngpu == 0, use cpu.")
args = parser.parse_args() args = parser.parse_args()
print_arguments(args, globals()) print_arguments(args, globals())
......
...@@ -43,7 +43,7 @@ if __name__ == "__main__": ...@@ -43,7 +43,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
'--nxpu', '--nxpu',
type=int, type=int,
default=1, default=0,
help="if nxpu == 0 and ngpu == 0, use cpu.") help="if nxpu == 0 and ngpu == 0, use cpu.")
parser.add_argument( parser.add_argument(
"--enable-auto-log", action="store_true", help="use auto log") "--enable-auto-log", action="store_true", help="use auto log")
......
...@@ -36,7 +36,7 @@ if __name__ == "__main__": ...@@ -36,7 +36,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
'--nxpu', '--nxpu',
type=int, type=int,
default=1, default=0,
help="if nxpu == 0 and ngpu == 0, use cpu.") help="if nxpu == 0 and ngpu == 0, use cpu.")
args = parser.parse_args() args = parser.parse_args()
print("model_type:{}".format(args.model_type)) print("model_type:{}".format(args.model_type))
......
...@@ -119,6 +119,9 @@ class Trainer(): ...@@ -119,6 +119,9 @@ class Trainer():
paddle.set_device('xpu') paddle.set_device('xpu')
elif self.args.ngpu > 0: elif self.args.ngpu > 0:
paddle.set_device("gpu") paddle.set_device("gpu")
else:
raise Exception("invalid device")
if self.parallel: if self.parallel:
self.init_parallel() self.init_parallel()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册