fix yolov3 cpu (#4122)

76f0eeae · Kaipeng Deng · GitHub · 9e0ab0b2 · 76f0eeae
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

PaddleCV/yolov3/train.py PaddleCV/yolov3/train.py +2 -2

未找到文件。
--- a/PaddleCV/yolov3/train.py
+++ b/PaddleCV/yolov3/train.py
@@ -77,8 +77,8 @@ def train():
    loss = model.loss()
    loss.persistable = True
-    devices_num = get_device_num()
+    devices_num = get_device_num() if cfg.use_gpu else 1
-    print("Found {} CUDA devices.".format(devices_num))
+    print("Found {} CUDA/CPU devices.".format(devices_num))
    learning_rate = cfg.learning_rate
    boundaries = cfg.lr_steps