From a327369a241b0cf838eed0eb841a79f112195454 Mon Sep 17 00:00:00 2001 From: lilong12 Date: Wed, 16 Jun 2021 19:00:33 +0800 Subject: [PATCH] bug fix, test=develop (#33594) --- .../paddle/distributed/fleet/meta_optimizers/sharding/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py b/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py index 85f114d7f71..c10978e9d94 100755 --- 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 -- GitLab