diff --git a/fluid/PaddleRec/ctr/reader.py b/fluid/PaddleRec/ctr/reader.py
index affd1d0e722ab957bd990b01702d0787c84470bd..e6bcc11d4f465d48f99ef2aa8059a952a3f8dfd4 100644
--- a/fluid/PaddleRec/ctr/reader.py
+++ b/fluid/PaddleRec/ctr/reader.py
@@ -46,7 +46,7 @@ class CriteoDataset(Dataset):
         return self._reader_creator(file_list, True, trainer_num, trainer_id)
 
     def test(self, file_list):
-        return self._reader_creator(file_list, False, -1)
+        return self._reader_creator(file_list, False, 1, 0)
 
     def infer(self, file_list):
-        return self._reader_creator(file_list, False, -1)
+        return self._reader_creator(file_list, False, 1, 0)