提交 cc17334b 编写于 作者: B baiyfbupt

disable multi-process

上级 ec96bbf2
...@@ -213,20 +213,17 @@ def main(): ...@@ -213,20 +213,17 @@ def main():
capacity=512, capacity=512,
use_double_buffer=True, use_double_buffer=True,
iterable=True, iterable=True,
return_list=True, return_list=True)
use_multiprocess=True)
valid_loader = fluid.io.DataLoader.from_generator( valid_loader = fluid.io.DataLoader.from_generator(
capacity=512, capacity=512,
use_double_buffer=True, use_double_buffer=True,
iterable=True, iterable=True,
return_list=True, return_list=True)
use_multiprocess=True)
dev_loader = fluid.io.DataLoader.from_generator( dev_loader = fluid.io.DataLoader.from_generator(
capacity=512, capacity=512,
use_double_buffer=True, use_double_buffer=True,
iterable=True, iterable=True,
return_list=True, return_list=True)
use_multiprocess=True)
train_loader.set_batch_generator(train_reader, places=place) train_loader.set_batch_generator(train_reader, places=place)
valid_loader.set_batch_generator(valid_reader, places=place) valid_loader.set_batch_generator(valid_reader, places=place)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册