From 32fce19a94ab40ad42aec9862ead55b04421b624 Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Wed, 4 Sep 2019 19:23:25 +0800
Subject: [PATCH] Fix devices issue

---
 pdseg/eval.py  | 2 +-
 pdseg/train.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pdseg/eval.py b/pdseg/eval.py
index c34a8b5e..815ebf56 100644
--- a/pdseg/eval.py
+++ b/pdseg/eval.py
@@ -102,7 +102,7 @@ def evaluate(cfg, ckpt_dir=None, use_gpu=False, use_mpio=False, **kwargs):
     places = fluid.cuda_places() if use_gpu else fluid.cpu_places()
     place = places[0]
     dev_count = len(places)
-    print("Device count = {}".format(dev_count))
+    print("#Device count: {}".format(dev_count))
 
     exe = fluid.Executor(place)
     exe.run(startup_prog)
diff --git a/pdseg/train.py b/pdseg/train.py
index 05db0de8..8738e43c 100644
--- a/pdseg/train.py
+++ b/pdseg/train.py
@@ -216,7 +216,7 @@ def train(cfg):
     place = places[0]
     # Get number of GPU
     dev_count = len(places)
-    print("#GPU-Devices: {}".format(dev_count))
+    print("#Device count: {}".format(dev_count))
 
     # Make sure BATCH_SIZE can divided by GPU cards
     assert cfg.BATCH_SIZE % dev_count == 0, (
-- 
GitLab