From 59a25fd2c9178beb057d70c9314ead56214500a3 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Tue, 29 May 2018 16:55:34 +0800 Subject: [PATCH] update --- python/paddle/fluid/transpiler/distribute_transpiler.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 1733c974285..e9b7d9e9d2d 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -410,7 +410,8 @@ class DistributeTranspiler: attrs={"axis": 0}) if self.has_distributed_lookup_table: - self._replace_lookup_table_op_with_prefetch(program, pserver_endpoints) + self._replace_lookup_table_op_with_prefetch(program, + pserver_endpoints) self._split_table_grad_and_add_send_vars(program, pserver_endpoints) def get_trainer_program(self): @@ -631,7 +632,8 @@ class DistributeTranspiler: return s_prog # transpiler function for dis lookup_table - def _replace_lookup_table_op_with_prefetch(self, program, pserver_endpoints): + def _replace_lookup_table_op_with_prefetch(self, program, + pserver_endpoints): # 1. replace lookup_table_op with split_ids_op -> prefetch_op -> sum_op self.prefetch_input_vars = None self.prefetch_output_vars = None -- GitLab