未验证 提交 3300a532 编写于 作者: Y Yu Yang 提交者: GitHub

Merge pull request #12546 from reyoung/feature/fix_nmt_cannot_save_checkpoint

Do not set loss@Grad as persistable
...@@ -572,8 +572,6 @@ def append_backward(loss, parameter_list=None, no_grad_set=None, ...@@ -572,8 +572,6 @@ def append_backward(loss, parameter_list=None, no_grad_set=None,
program.current_block_idx = current_block_idx program.current_block_idx = current_block_idx
program._sync_with_cpp() program._sync_with_cpp()
# FIXME(zcd): prevent loss.grad optimized by mem_opt.
loss.block.var(_append_grad_suffix_(loss.name)).persistable = True
if parameter_list is not None: if parameter_list is not None:
parameters = parameter_list parameters = parameter_list
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册