m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix dy-load bug, test=develop

Browse Source
shanyi15-patch-1
lujun 6 years ago
parent a3f17280a3
commit 94c4cd1a98
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File

3
python/paddle/fluid/dygraph/checkpoint.py
Unescape View File

@ -133,7 +133,6 @@ def _save_var_to_file(stat_dict, file_dir, file_name):
def _load_var_from_file(file_dir): def _load_var_from_file(file_dir):
def walk_filename(file_dir): def walk_filename(file_dir):
var_name_list = [] var_name_list = []
if os.path.exists(file_dir) and os.path.exists(os.path.join(file_dir)): if os.path.exists(file_dir) and os.path.exists(os.path.join(file_dir)):
@ -148,9 +147,7 @@ def _load_var_from_file(file_dir):
load_block = default_main_program().global_block() load_block = default_main_program().global_block()
load_var_map = {} load_var_map = {}
file_var_list = walk_filename(file_dir) file_var_list = walk_filename(file_dir)
for var_name in file_var_list: for var_name in file_var_list:
new_var = Variable(block=load_block, name=var_name) new_var = Variable(block=load_block, name=var_name)
load_block.append_op( load_block.append_op(

Write Preview
Loading…
Cancel
Save