add optimized_guard for ModelAverage

guochaorong-patch-1
qiaolongfei 7 years ago
parent 971cf70517
commit 2ce1ed3dbd

@ -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()

Loading…
Cancel
Save