提交 23441fa3 编写于 作者: X xiaoting 提交者: qingqing01

fix typo for faster_rcnn_50_2x.yml (#2515)

* Fix type for faster_rcnn_50_2x.
* Make sure that type of cpu_num is int.
上级 5b584500
...@@ -43,7 +43,7 @@ RPNHead: ...@@ -43,7 +43,7 @@ RPNHead:
train_proposal: train_proposal:
min_size: 0.0 min_size: 0.0
nms_thresh: 0.7 nms_thresh: 0.7
pre_nums_top_n: 12000 pre_nms_top_n: 12000
post_nms_top_n: 2000 post_nms_top_n: 2000
test_proposal: test_proposal:
min_size: 0.0 min_size: 0.0
......
...@@ -51,7 +51,7 @@ def main(): ...@@ -51,7 +51,7 @@ def main():
if cfg['use_gpu']: if cfg['use_gpu']:
devices_num = fluid.core.get_cuda_device_count() devices_num = fluid.core.get_cuda_device_count()
else: else:
devices_num = os.environ.get('CPU_NUM', multiprocessing.cpu_count()) devices_num = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count()))
if 'eval_feed' not in cfg: if 'eval_feed' not in cfg:
eval_feed = create(main_arch + 'EvalFeed') eval_feed = create(main_arch + 'EvalFeed')
......
...@@ -53,7 +53,7 @@ def main(): ...@@ -53,7 +53,7 @@ def main():
if cfg['use_gpu']: if cfg['use_gpu']:
devices_num = fluid.core.get_cuda_device_count() devices_num = fluid.core.get_cuda_device_count()
else: else:
devices_num = os.environ.get('CPU_NUM', multiprocessing.cpu_count()) devices_num = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count()))
if 'train_feed' not in cfg: if 'train_feed' not in cfg:
train_feed = create(type(main_arch).__name__ + 'TrainFeed') train_feed = create(type(main_arch).__name__ + 'TrainFeed')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册