diff --git a/python/paddle/fluid/dygraph/checkpoint.py b/python/paddle/fluid/dygraph/checkpoint.py index ee3bbd938096ed5c7d86bad260af2936a873126c..3d797d5049ea35e4af7da68ab47d6131d8e6e3f5 100644 --- a/python/paddle/fluid/dygraph/checkpoint.py +++ b/python/paddle/fluid/dygraph/checkpoint.py @@ -133,7 +133,6 @@ def _save_var_to_file(stat_dict, file_dir, file_name): def _load_var_from_file(file_dir): - def walk_filename(file_dir): var_name_list = [] 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_var_map = {} - file_var_list = walk_filename(file_dir) - for var_name in file_var_list: new_var = Variable(block=load_block, name=var_name) load_block.append_op(