diff --git a/fluid/PaddleRec/ctr/reader.py b/fluid/PaddleRec/ctr/reader.py index affd1d0e722ab957bd990b01702d0787c84470bd..6c58d1286b6331b97a09e828c28581e769b246f1 100644 --- a/fluid/PaddleRec/ctr/reader.py +++ b/fluid/PaddleRec/ctr/reader.py @@ -13,7 +13,7 @@ class CriteoDataset(Dataset): self.continuous_range_ = range(1, 14) self.categorical_range_ = range(14, 40) - def _reader_creator(self, file_list, is_train, trainer_num, trainer_id): + def _reader_creator(self, file_list, is_train, trainer_num, trainer_id=0): def reader(): for file in file_list: with open(file, 'r') as f: