提交 744bc0bf 编写于 作者: C chengduozh

use distributed_batch_reader

上级 0f65b4dd
...@@ -280,7 +280,7 @@ def main(args): ...@@ -280,7 +280,7 @@ def main(args):
main_program=train_program) 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.multi_process_reader( train_data_generator = fluid.contrib.reader.distributed_batch_reader(
train_data_generator) train_data_generator)
train_pyreader.decorate_tensor_provider(train_data_generator) train_pyreader.decorate_tensor_provider(train_data_generator)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册