提交 2ce1ed3d 编写于 作者: Q qiaolongfei

add optimized_guard for ModelAverage

上级 971cf705
...@@ -1158,6 +1158,7 @@ class ModelAverage(Optimizer): ...@@ -1158,6 +1158,7 @@ class ModelAverage(Optimizer):
self.params_grads.append((param, grad)) self.params_grads.append((param, grad))
for param, grad in self.params_grads: for param, grad in self.params_grads:
with param.block.program.optimized_guard(param):
self._append_average_accumulate_op(param) self._append_average_accumulate_op(param)
self.apply_program = Program() self.apply_program = Program()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册