remove distributed_batch_reader

37867372 · chengduozh · 9e92cb5d · 37867372
显示空白变更内容
内联并排

Showing with 4 addition and 4 deletion

BERT/run_classifier.py BERT/run_classifier.py +4 -4

未找到文件。
--- 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))
+        # num_trainers = int(os.environ.get('PADDLE_TRAINERS_NUM', 1))
-        if num_trainers > 1:
+        # if num_trainers > 1:
-            train_data_generator = fluid.contrib.reader.distributed_batch_reader(
+        #     train_data_generator = fluid.contrib.reader.distributed_batch_reader(
-                train_data_generator)
+        #         train_data_generator)
        train_pyreader.decorate_tensor_provider(train_data_generator)
    else: