diff --git a/ppocr/utils/utility.py b/ppocr/utils/utility.py index 2cf3c8f5c9ebba07ee1c21fe2248fe3f600126d9..e27dd1d8738a25c6a6669b99ad2b6eed4a9f25d0 100755 --- a/ppocr/utils/utility.py +++ b/ppocr/utils/utility.py @@ -90,3 +90,15 @@ def check_and_read_gif(img_path): return imgvalue, True return None, False + +def create_multi_devices_program(program, loss_var_name): + build_strategy = fluid.BuildStrategy() + build_strategy.memory_optimize = False + build_strategy.enable_inplace = True + exec_strategy = fluid.ExecutionStrategy() + exec_strategy.num_iteration_per_drop_scope = 1 + compile_program = fluid.CompiledProgram(program).with_data_parallel( + loss_name=loss_var_name, + build_strategy=build_strategy, + exec_strategy=exec_strategy) + return compile_program