diff --git a/fluid/PaddleRec/ctr/reader.py b/fluid/PaddleRec/ctr/reader.py index c733af04f8e5ddeb2a3042fe8bf67397a752d229..851839c35b79dea87f51d5aeb5eb1491b7670377 100644 --- a/fluid/PaddleRec/ctr/reader.py +++ b/fluid/PaddleRec/ctr/reader.py @@ -21,10 +21,10 @@ class CriteoDataset(Dataset): for line in f: line_idx += 1 if is_train and line_idx > self.train_idx_: - continue + break elif not is_train and line_idx <= self.train_idx_: continue - if trainer_id > 0 and line_idx % trainer_num != trainer_id: + if line_idx % trainer_num != trainer_id: continue features = line.rstrip('\n').split('\t') dense_feature = [] diff --git a/fluid/PaddleRec/ctr/train.py b/fluid/PaddleRec/ctr/train.py index 71f2aed7e78989c3e02d24df49bd03411a1e9a52..8aa5c23f231b0b68c65f6080ab9ab43bdcfa848b 100644 --- a/fluid/PaddleRec/ctr/train.py +++ b/fluid/PaddleRec/ctr/train.py @@ -138,7 +138,7 @@ def train(): if args.is_local: logger.info("run local training") main_program = fluid.default_main_program() - train_loop(args, main_program, data_list, loss, auc_var, batch_auc_var, 1, -1) + train_loop(args, main_program, data_list, loss, auc_var, batch_auc_var, 1, 0) else: logger.info("run dist training") t = fluid.DistributeTranspiler() @@ -154,7 +154,7 @@ def train(): logger.info("run trainer") train_prog = t.get_trainer_program() train_loop(args, train_prog, data_list, loss, auc_var, batch_auc_var, - args.trainers, args.trainer_id + 1) + args.trainers, args.trainer_id) if __name__ == '__main__':