From 378673722a7664e8267ea22425bad4fd15b1053e Mon Sep 17 00:00:00 2001 From: chengduozh Date: Wed, 19 Jun 2019 14:58:21 +0800 Subject: [PATCH] remove distributed_batch_reader --- BERT/run_classifier.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/BERT/run_classifier.py b/BERT/run_classifier.py index 353cbec..da15071 100644 --- a/BERT/run_classifier.py +++ b/BERT/run_classifier.py @@ -279,10 +279,10 @@ def main(args): exec_strategy=exec_strategy, build_strategy = build_strategy, main_program=train_program) - num_trainers = int(os.environ.get('PADDLE_TRAINERS_NUM', 1)) - if num_trainers > 1: - train_data_generator = fluid.contrib.reader.distributed_batch_reader( - train_data_generator) + # num_trainers = int(os.environ.get('PADDLE_TRAINERS_NUM', 1)) + # if num_trainers > 1: + # train_data_generator = fluid.contrib.reader.distributed_batch_reader( + # train_data_generator) train_pyreader.decorate_tensor_provider(train_data_generator) else: -- GitLab