Skip to content

Commit 383bde0

Browse files
authored
fix multi-gpu training in static (PaddlePaddle#4835)
1 parent 511d179 commit 383bde0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

static/tools/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def main():
112112

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

0 commit comments

Comments
 (0)