fix cascade_rcnn_cls_aware_r101_vd_fpn_ms_test (#1051)

499dafa8 · wangguanzhong · GitHub · e60d5598 · 499dafa8
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml +5 -0

未找到文件。
--- a/configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml
+++ b/configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml
@@ -109,6 +109,11 @@ OptimizerBuilder:
    factor: 0.0001
    type: L2
+_READER_: 'faster_fpn_reader.yml'
+TrainReader:
+  batch_size: 2
 EvalReader:
  batch_size: 1
  inputs_def: