diff --git a/PaddleCV/object_detection/train.py b/PaddleCV/object_detection/train.py index 6fb7ce0236dc63f39de597788bc425f8dfa5ae6d..0b660a94f201ce05a0273e8c0bf7d4c727620944 100644 --- a/PaddleCV/object_detection/train.py +++ b/PaddleCV/object_detection/train.py @@ -179,8 +179,11 @@ def train(args, predicate=if_exist) if parallel: + build_strategy = fluid.BuildStrategy() + build_strategy.enable_inplace = True + build_strategy.memory_optimize = True train_exe = fluid.ParallelExecutor(main_program=train_prog, - use_cuda=use_gpu, loss_name=loss.name) + use_cuda=use_gpu, loss_name=loss.name, build_strategy=build_strategy) train_reader = reader.train(data_args, train_file_list, batch_size_per_device,