Skip to content

Commit f6b06bd

Browse files
committed
Merge branch 'dis_ckpt_fix' of github.com:seiriosPlus/Paddle into dis_ckpt_fix
2 parents 44bade8 + 7dbf5f1 commit f6b06bd

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

python/paddle/fluid/io.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,7 @@ def name_has_fc(var):
408408

409409
# load slice vars on pserver, if have it.
410410
_load_slice_up_vars(executor, dirname,
411-
main_program._slice_vars_and_atts)
411+
main_program._slice_vars_and_attrs)
412412

413413

414414
def load_params(executor, dirname, main_program=None, filename=None):
@@ -871,14 +871,14 @@ def get_parameter_value_by_name(name, executor, program=None):
871871
return get_parameter_value(var, executor)
872872

873873

874-
def _load_slice_up_vars(executor, dirname, slice_vars_and_atts):
875-
if not slice_vars_and_atts:
874+
def _load_slice_up_vars(executor, dirname, slice_vars_and_attrs):
875+
if not slice_vars_and_attrs:
876876
return
877877

878878
load_prog = Program()
879879
load_block = load_prog.global_block()
880880

881-
for var_tuple in slice_vars_and_atts:
881+
for var_tuple in slice_vars_and_attrs:
882882
orig_var = var_tuple[0]
883883
start = var_tuple[1]
884884
slice_var = var_tuple[2]

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -712,7 +712,7 @@ def _get_slice_vars_and_attrs(self, endpoint):
712712
slice_vars_and_attrs = []
713713
block_suffix = "block"
714714
for param in self.param_grad_ep_mapping[endpoint]["params"]:
715-
orig_var_name, block_name, _ = self._get_varname_parts(param)
715+
orig_var_name, block_name, _ = self._get_varname_parts(param.name)
716716
if not block_name:
717717
continue
718718

0 commit comments

Comments
 (0)