diff --git a/benchmark/fluid/models/resnet.py b/benchmark/fluid/models/resnet.py
index 98a7d76dfd5f85db070977ea822c7510b6d2eed1..9ed1093c54a501cc93dbbf9c3651fe70914ce26b 100644
--- a/benchmark/fluid/models/resnet.py
+++ b/benchmark/fluid/models/resnet.py
@@ -201,8 +201,8 @@ def get_model(args):
             train_reader, buf_size=5120),
         batch_size=args.batch_size * args.gpus,
         drop_last=True)
-    batched_test_reader = paddle.batch(train_reader,
-               batch_size=args.batch_size, drop_last=True)
+    batched_test_reader = paddle.batch(
+        train_reader, batch_size=args.batch_size, drop_last=True)
 
     return avg_cost, inference_program, optimizer, batched_train_reader,\
                    batched_test_reader, batch_acc