From 499dafa8e8f5f8eeb149f2fb3a3ea0e1f747ca13 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Fri, 10 Jul 2020 15:07:15 +0800
Subject: [PATCH] fix cascade_rcnn_cls_aware_r101_vd_fpn_ms_test (#1051)

---
 configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml b/configs/cascade_rcnn_cls_aware_r101_vd_fpn_ms_test.yml
index c03c5474d..e7deb4825 100644
--- 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:
-- 
GitLab