未验证 提交 9a2d327c 编写于 作者: Y Yuang Liu 提交者: GitHub

bug fix for adamw (#37905)

上级 b8793f70
......@@ -6582,7 +6582,10 @@ class RecomputeOptimizer(Optimizer):
print("Finished apply_optimize")
"""
return self._optimizer.apply_optimize(
func = self._optimizer.apply_optimize if hasattr(
self._optimizer,
'apply_optimize') else self._optimizer._apply_optimize
return func(
loss, startup_program=startup_program, params_grads=params_grads)
def minimize(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册