fix ctr test reader

5f1fd923 · Qiao Longfei · e3c3af79 · 5f1fd923
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

fluid/PaddleRec/ctr/reader.py fluid/PaddleRec/ctr/reader.py +2 -2

未找到文件。
--- 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)