fix multi-gpu training in static (#4835)

383bde0b · wangguanzhong · GitHub · 511d179b · 383bde0b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

static/tools/train.py static/tools/train.py +1 -1

未找到文件。
--- a/static/tools/train.py
+++ b/static/tools/train.py
@@ -112,7 +112,7 @@ def main():

    if cfg.use_gpu:
        devices_num = fluid.core.get_cuda_device_count()
-    if cfg.use_npu:
+    elif cfg.use_npu:
        devices_num = fluid.core.get_npu_device_count()
    elif use_xpu:
        # ToDo(qingshu): XPU only support single card now