disable multi-process

cc17334b · baiyfbupt · ec96bbf2 · cc17334b
隐藏空白更改
内联并排

Showing with 3 addition and 6 deletion

demo/bert/train_cell_base.py demo/bert/train_cell_base.py +3 -6

未找到文件。
--- a/demo/bert/train_cell_base.py
+++ b/demo/bert/train_cell_base.py
@@ -213,20 +213,17 @@ def main():
            capacity=512,
            use_double_buffer=True,
            iterable=True,
-            return_list=True,
-            use_multiprocess=True)
+            return_list=True)
        valid_loader = fluid.io.DataLoader.from_generator(
            capacity=512,
            use_double_buffer=True,
            iterable=True,
-            return_list=True,
-            use_multiprocess=True)
+            return_list=True)
        dev_loader = fluid.io.DataLoader.from_generator(
            capacity=512,
            use_double_buffer=True,
            iterable=True,
-            return_list=True,
-            use_multiprocess=True)
+            return_list=True)

        train_loader.set_batch_generator(train_reader, places=place)
        valid_loader.set_batch_generator(valid_reader, places=place)