未验证 提交 c0a6ebd1 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix label_smooth setting (#2884)

上级 07a926fb
......@@ -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
......
......@@ -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
......
......@@ -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:
......
......@@ -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():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册