diff --git a/BERT/run_classifier.py b/BERT/run_classifier.py index 353cbec37e537cbd5c176628344304919ef40622..da15071b1c0d37173cf524060215a04fcd96bd02 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: