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

bug fix, test=develop (#33594)

上级 debae94e
......@@ -561,7 +561,7 @@ def save_persistables(exe, dirname, main_program, filename=None):
"""
# TODO (JZ-LIANG) revise this for uniform mixed parallelism
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):
# 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.
先完成此消息的编辑!
想要评论请 注册