@ -1158,6 +1158,7 @@ class ModelAverage(Optimizer):
self.params_grads.append((param, grad))
for param, grad in self.params_grads:
with param.block.program.optimized_guard(param):
self._append_average_accumulate_op(param)
self.apply_program = Program()