diff --git a/PaddleCV/yolov3/README.md b/PaddleCV/yolov3/README.md index 6ca1c3ebf65c4a1e02f14c5c6217ae636349fcc7..5d21978e443b1018f53b87b0141b6a2de0555479 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 64fe811a3bf54f74368902c52abc554818ed73a7..a022240ce9fac530f55490e6155c332eebf6b8a6 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 fe49124947030b9b2ded7e1b5361ba84ba876324..c84606252c7e29f581ba853c2e93a5e809c6f9d0 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 aef5e2fa6b557e5608d2700e1d1f8a7ded7a5ffd..56911b262845265fdb2df539e298052b90188f8b 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():