Merge pull request #12445 from jacquesqiao/fix-dis-trans

Fix dist trans
7 years ago · 271b72460f
parent 059b27840c f761874b22
commit 271b72460f
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@ -495,6 +495,7 @@ class DistributeTranspiler(object):
            pserver_index = self.pserver_endpoints.index(endpoint)
            table_opt_block = self._create_table_optimize_block(
                pserver_index, pserver_program, pre_block_idx, grad_to_block_id)
+            optimize_blocks.append(table_opt_block)
            prefetch_var_name_to_block_id = self._create_prefetch_block(
                pserver_index, pserver_program, table_opt_block)
            checkpoint_block_id = self._create_checkpoint_save_block(