Fix bug in continuous apply, test=develop (#27337)

4 years ago · 34091533c9
parent f992f8d7ef
commit 34091533c9
1 changed files with 4 additions and 2 deletions
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@ -3570,8 +3570,10 @@ class ExponentialMovingAverage(object):
                # bias correction
                with layers.control_flow.Switch() as switch:
                    with switch.case(global_step > 0):
-                        layers.assign(output=ema, input=ema / (1.0 - decay_pow))
-                layers.assign(input=ema, output=param)
+                        layers.assign(
+                            output=param, input=ema / (1.0 - decay_pow))
+                    with switch.default():
+                        layers.assign(output=param, input=ema)

        self.restore_program = Program()
        block = self.restore_program.global_block()