diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index 4d4fb32c8c3eab26f0a26a504bfb16115db89096..e094ba8f268e4087cf5ce3824f88375249517b22 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -408,7 +408,7 @@ def load_vars(executor, # load slice vars on pserver, if have it. _load_slice_up_vars(executor, dirname, - main_program._slice_vars_and_atts) + main_program._slice_vars_and_attrs) def load_params(executor, dirname, main_program=None, filename=None): @@ -871,14 +871,14 @@ def get_parameter_value_by_name(name, executor, program=None): return get_parameter_value(var, executor) -def _load_slice_up_vars(executor, dirname, slice_vars_and_atts): - if not slice_vars_and_atts: +def _load_slice_up_vars(executor, dirname, slice_vars_and_attrs): + if not slice_vars_and_attrs: return load_prog = Program() load_block = load_prog.global_block() - for var_tuple in slice_vars_and_atts: + for var_tuple in slice_vars_and_attrs: orig_var = var_tuple[0] start = var_tuple[1] slice_var = var_tuple[2] diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 5cc447f1dd8736fd88bc72890bd46ea383e3491a..5da6353e6892d05363e318a8a393d1b788631ae3 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -712,7 +712,7 @@ class DistributeTranspiler(object): slice_vars_and_attrs = [] block_suffix = "block" for param in self.param_grad_ep_mapping[endpoint]["params"]: - orig_var_name, block_name, _ = self._get_varname_parts(param) + orig_var_name, block_name, _ = self._get_varname_parts(param.name) if not block_name: continue