未验证 提交 172f2719 编写于 作者: L lilong12 提交者: GitHub

bug fix, test=develop (#33595)

上级 5c68e79d
...@@ -555,7 +555,7 @@ def save_persistables(exe, dirname, main_program, filename=None): ...@@ -555,7 +555,7 @@ def save_persistables(exe, dirname, main_program, filename=None):
""" """
# TODO (JZ-LIANG) revise this for uniform mixed parallelism # TODO (JZ-LIANG) revise this for uniform mixed parallelism
if main_program._pipeline_opt: if main_program._pipeline_opt:
main_program = main_program._pipeline_opt['section_program']['program'] main_program = main_program._pipeline_opt['section_program']
def is_opt_vars(var): def is_opt_vars(var):
# NOTE(JZ-LIANG): The checks should be updated when add new compatible optimizer # NOTE(JZ-LIANG): The checks should be updated when add new compatible optimizer
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册