Skip to content

Commit d0beac3

Browse files
MoFHekarhdong
authored andcommitted
[fix] Unable to load KV files in FileSystem Saver when using name_scope, such as optimizer variable.
1 parent 59966f2 commit d0beac3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/dynamic_embedding_variable.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,8 +174,10 @@ def _list_de_variable_saved_files_from_file_system(de_variable_name,
174174
else:
175175
keys_pattern = '_mht_*of*_rank*_size{}-keys'.format(proc_size)
176176
values_pattern = '_mht_*of*_rank*_size{}-values'.format(proc_size)
177+
de_variable_saveable_name = string_ops.regex_replace(de_variable_name, "/",
178+
"_")
177179
_shard_name_base_dir = string_ops.string_join(
178-
[de_variable_folder_path, de_variable_name], separator='/')
180+
[de_variable_folder_path, de_variable_saveable_name], separator='/')
179181
_shard_name_keys_pattern = string_ops.string_join(
180182
[_shard_name_base_dir, keys_pattern], separator='')
181183
shard_keys_file_list = gen_io_ops.matching_files(_shard_name_keys_pattern)

0 commit comments

Comments
 (0)