Skip to content

Commit d6d18ad

Browse files
committed
fix utest bug at windows, test=release/1.4
1 parent e3e2b62 commit d6d18ad

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

python/paddle/fluid/dygraph/checkpoint.py

+14-4
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,10 @@ def _save_var_to_file(stat_dict, file_dir, file_name):
120120
type='save',
121121
inputs={'X': [each_var]},
122122
outputs={},
123-
attrs={'file_path': os.path.join(file_dir, each_var.name)})
123+
attrs={
124+
'file_path': os.path.join(file_dir,
125+
os.path.normpath(each_var.name))
126+
})
124127

125128
if file_name is not None:
126129
save_var_list = []
@@ -131,7 +134,9 @@ def _save_var_to_file(stat_dict, file_dir, file_name):
131134
type='save_combine',
132135
inputs={'X': save_var_list},
133136
outputs={},
134-
attrs={'file_path': os.path.join(file_dir, file_name)})
137+
attrs={
138+
'file_path': os.path.join(file_dir, os.path.normpath(file_name))
139+
})
135140

136141

137142
def _load_var_from_file(stat_dict, file_dir, file_name):
@@ -148,7 +153,10 @@ def _load_var_from_file(stat_dict, file_dir, file_name):
148153
type='load',
149154
inputs={},
150155
outputs={'Out': [new_var]},
151-
attrs={'file_path': os.path.join(file_dir, each_var.name)})
156+
attrs={
157+
'file_path': os.path.join(file_dir,
158+
os.path.normpath(each_var.name))
159+
})
152160

153161
load_var_map[new_var.name] = new_var
154162

@@ -161,7 +169,9 @@ def _load_var_from_file(stat_dict, file_dir, file_name):
161169
type='load_combine',
162170
inputs={},
163171
outputs={"Out": load_var_list},
164-
attrs={'file_path': os.path.join(file_dir, file_name)})
172+
attrs={
173+
'file_path': os.path.join(file_dir, os.path.normpath(file_name))
174+
})
165175
for res_var in load_var_list:
166176
load_var_map[res_var.name] = res_var
167177

0 commit comments

Comments
 (0)