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 c03c5474d1388383a5958a9fa71895c8c73117f0..e7deb4825ef7ad111d8bbd62c46bac894ee3fef9 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: