From 8fb6cc5382b69f08046fa8dc9829f29fa1606083 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Fri, 28 Oct 2022 11:08:04 +0000
Subject: [PATCH] fix: debug

---
 ppcls/static/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppcls/static/train.py b/ppcls/static/train.py
index 86e83249..c58aaf52 100644
--- a/ppcls/static/train.py
+++ b/ppcls/static/train.py
@@ -93,8 +93,8 @@ def main(args):
     use_npu = global_config.get("use_npu", False)
     use_mlu = global_config.get("use_mlu", False)
     assert (
-        use_gpu and use_xpu and use_npu and use_mlu
-    ) is not True, "gpu, xpu, npu and mlu can not be true in the same time in static mode!"
+        use_gpu + use_xpu + use_npu + use_mlu + use_ascend <= 1
+    ), "gpu, xpu, npu, mlu and ascend can not be true in the same time in static mode!"
 
     if use_gpu:
         device = paddle.set_device('gpu')
-- 
GitLab