提交 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, ...@@ -408,7 +408,7 @@ def load_vars(executor,
# load slice vars on pserver, if have it. # load slice vars on pserver, if have it.
_load_slice_up_vars(executor, dirname, _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): def load_params(executor, dirname, main_program=None, filename=None):
...@@ -871,14 +871,14 @@ def get_parameter_value_by_name(name, executor, program=None): ...@@ -871,14 +871,14 @@ def get_parameter_value_by_name(name, executor, program=None):
return get_parameter_value(var, executor) return get_parameter_value(var, executor)
def _load_slice_up_vars(executor, dirname, slice_vars_and_atts): def _load_slice_up_vars(executor, dirname, slice_vars_and_attrs):
if not slice_vars_and_atts: if not slice_vars_and_attrs:
return return
load_prog = Program() load_prog = Program()
load_block = load_prog.global_block() 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] orig_var = var_tuple[0]
start = var_tuple[1] start = var_tuple[1]
slice_var = var_tuple[2] slice_var = var_tuple[2]
......
...@@ -712,7 +712,7 @@ class DistributeTranspiler(object): ...@@ -712,7 +712,7 @@ class DistributeTranspiler(object):
slice_vars_and_attrs = [] slice_vars_and_attrs = []
block_suffix = "block" block_suffix = "block"
for param in self.param_grad_ep_mapping[endpoint]["params"]: 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: if not block_name:
continue continue
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册