From 172f27191002b21a31c1cbb2df092e4446b67606 Mon Sep 17 00:00:00 2001 From: lilong12 Date: Wed, 16 Jun 2021 22:01:40 +0800 Subject: [PATCH] bug fix, test=develop (#33595) --- .../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 f4ceb2d287a..a628105de0f 100755 --- a/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py +++ b/python/paddle/distributed/fleet/meta_optimizers/sharding/utils.py @@ -555,7 +555,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