bug fix, test=develop (#33594)

a327369a · lilong12 · GitHub · debae94e · a327369a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py ...addle/distributed/fleet/meta_optimizers/sharding/utils.py +1 -1

未找到文件。
--- a/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py
@@ -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