From 34091533c95291dd96459af65dacf04a9cce9fa2 Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Wed, 16 Sep 2020 16:19:17 +0800 Subject: [PATCH] Fix bug in continuous apply, test=develop (#27337) --- python/paddle/fluid/optimizer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py index 192effd2e4..1e7915ed78 100755 --- 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()