From f5ce79e8f5c91f50a2b06ea2129c19b68ebef76f Mon Sep 17 00:00:00 2001
From: Kaipeng Deng <dengkaipeng@baidu.com>
Date: Mon, 22 Jul 2019 17:24:42 +0800
Subject: [PATCH] fix label_smooth setting (#2885)

---
 PaddleCV/yolov3/README.md        | 2 +-
 PaddleCV/yolov3/README_en.md     | 2 +-
 PaddleCV/yolov3/models/yolov3.py | 2 +-
 PaddleCV/yolov3/reader.py        | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/PaddleCV/yolov3/README.md b/PaddleCV/yolov3/README.md
index 6ca1c3eb..5d21978e 100644
--- a/PaddleCV/yolov3/README.md
+++ b/PaddleCV/yolov3/README.md
@@ -94,7 +94,7 @@ dataset/coco/
        --class_num=${category_num}
 
 - 通过设置`export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7`指定8卡GPU训练。
-- 若在Windows环境下训练模型，建议设置`--use_multiprocess=False`。
+- 若在Windows环境下训练模型，建议设置`--use_multiprocess_reader=False`。
 - 可选参数见：
 
     python train.py --help
diff --git a/PaddleCV/yolov3/README_en.md b/PaddleCV/yolov3/README_en.md
index 64fe811a..a022240c 100644
--- a/PaddleCV/yolov3/README_en.md
+++ b/PaddleCV/yolov3/README_en.md
@@ -95,7 +95,7 @@ Please make sure that pre-trained model is downloaded and loaded correctly, othe
        --class_num=${category_num}
 
 - Set `export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7` to specifiy 8 GPUs to train. 
-- It is recommended to set `--use_multiprocess=False` when training on Windows.
+- It is recommended to set `--use_multiprocess_reader=False` when training on Windows.
 - For more help on arguments:
 
     python train.py --help
diff --git a/PaddleCV/yolov3/models/yolov3.py b/PaddleCV/yolov3/models/yolov3.py
index fe491249..c8460625 100644
--- a/PaddleCV/yolov3/models/yolov3.py
+++ b/PaddleCV/yolov3/models/yolov3.py
@@ -176,7 +176,7 @@ class YOLOv3(object):
                     class_num=cfg.class_num,
                     ignore_thresh=cfg.ignore_thresh,
                     downsample_ratio=self.downsample,
-                    use_label_smooth=cfg.label_smooth,
+                    use_label_smooth=bool(cfg.label_smooth),
                     name="yolo_loss" + str(i))
                 self.losses.append(fluid.layers.reduce_mean(loss))
             else:
diff --git a/PaddleCV/yolov3/reader.py b/PaddleCV/yolov3/reader.py
index aef5e2fa..56911b26 100644
--- a/PaddleCV/yolov3/reader.py
+++ b/PaddleCV/yolov3/reader.py
@@ -303,8 +303,8 @@ def train(size=416,
         return generator
     else:
         print("multiprocess is not fully compatible with Windows, "
-                "you can set --use_multiprocess=False if there are "
-                "errors incured by multiprocess")
+                "you can set --use_multiprocess_reader=False if there "
+                "are errors incured by multiprocess")
         print("multiprocess reader starting up, it takes a while...")
 
     def infinite_reader():
-- 
GitLab