parallel-do should not merge the gradient of parameter witch set stop_gradient=True

7 years ago · 5d305070ec
parent f0548679b3
commit 5d305070ec
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/layers/control_flow.py
+++ b/python/paddle/fluid/layers/control_flow.py
@ -294,7 +294,8 @@ class ParallelDo(object):

        params = list(set(params))

-        return [parent_block.var(name) for name in params]
+        param_list = [parent_block.var(name) for name in params]
+        return filter(lambda param: param.stop_gradient is False, param_list)

    def complete_op(self):
        main_program = self.helper.main_program