From 5c7d6a5594a48ed688523eb7256f9743a632d23d Mon Sep 17 00:00:00 2001 From: Yancey1989 Date: Wed, 20 Jun 2018 10:08:48 +0800 Subject: [PATCH] update --- python/paddle/fluid/transpiler/distribute_transpiler.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index a0f9334c009..e94abfc006c 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -489,7 +489,7 @@ class DistributeTranspiler: # we put the new sub block to new block to follow the block # hierarchy of the original blocks new_sub_block = program.create_block(new_block.idx) - skip_sub_blks(new_sub_block.idx) + skip_sub_blks.append(new_sub_block.idx) # clone vars for var in origin_block.vars: @@ -504,7 +504,6 @@ class DistributeTranspiler: # reset the block of op op.set_attr('sub_block', new_sub_block) - return new_sub_block.idx # append lr decay ops to the child block if exists lr_ops = self._get_lr_ops() -- GitLab