Skip to content

Commit 172f271

Browse files
author
lilong12
authored
bug fix, test=develop (#33595)
1 parent 5c68e79 commit 172f271

File tree

1 file changed

+1
-1
lines changed
  • python/paddle/distributed/fleet/meta_optimizers/sharding

1 file changed

+1
-1
lines changed

python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -555,7 +555,7 @@ def save_persistables(exe, dirname, main_program, filename=None):
555555
"""
556556
# TODO (JZ-LIANG) revise this for uniform mixed parallelism
557557
if main_program._pipeline_opt:
558-
main_program = main_program._pipeline_opt['section_program']['program']
558+
main_program = main_program._pipeline_opt['section_program']
559559

560560
def is_opt_vars(var):
561561
# NOTE(JZ-LIANG): The checks should be updated when add new compatible optimizer

0 commit comments

Comments
 (0)