use distributed_batch_reader

744bc0bf · chengduozh · 0f65b4dd · 744bc0bf
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

BERT/run_classifier.py BERT/run_classifier.py +1 -1

未找到文件。
--- a/BERT/run_classifier.py
+++ b/BERT/run_classifier.py
@@ -280,7 +280,7 @@ def main(args):
            main_program=train_program)
        num_trainers = int(os.environ.get('PADDLE_TRAINERS_NUM', 1))
        if num_trainers > 1:
-            train_data_generator = fluid.contrib.reader.multi_process_reader(
+            train_data_generator = fluid.contrib.reader.distributed_batch_reader(
                train_data_generator)
        train_pyreader.decorate_tensor_provider(train_data_generator)