提交 f6b06bdb 编写于 作者: T tangwei12

Merge branch 'dis_ckpt_fix' of github.com:seiriosPlus/Paddle into dis_ckpt_fix

......@@ -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]
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册