change judgment logic for multi device

2022-10-25 18:53:24 +08:00 · 2022-10-25 18:53:24 +08:00 · c032293a77
parent 0cec70bd22
commit c032293a77
1 changed files with 1 additions and 1 deletions
--- a/ppcls/static/train.py
+++ b/ppcls/static/train.py
@ -94,7 +94,7 @@ def main(args):
    use_mlu = global_config.get("use_mlu", False)
    use_ascend = global_config.get("use_ascend", False)
    assert (
-        use_gpu and use_xpu and use_npu and use_mlu and use_ascend
+        use_gpu + use_xpu + use_npu + use_mlu + use_ascend <= 1
    ) is not True, "gpu, xpu, npu, mlu and ascend can not be true in the same time in static mode!"

    if use_gpu: