Skip to content

Commit ac2533a

Browse files
PWZERrhdong
authored andcommitted
update _get_dynamic_embedding_save_ops and _get_dynamic_embedding_restore_ops
1 parent bbab571 commit ac2533a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/tf_save_restore_patch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ class _DynamicEmbeddingSaver(saver.Saver):
175175
def _get_dynamic_embedding_save_ops(self):
176176
save_ops = tf_utils.ListWrapper([])
177177
if not self._var_list:
178-
return save_ops
178+
return control_flow_ops.group(save_ops.as_list())
179179

180180
for var in self._var_list:
181181
de_var = None
@@ -202,7 +202,7 @@ def _get_dynamic_embedding_save_ops(self):
202202
def _get_dynamic_embedding_restore_ops(self):
203203
restore_ops = tf_utils.ListWrapper([])
204204
if not self._var_list:
205-
return restore_ops
205+
return control_flow_ops.group(restore_ops.as_list())
206206

207207
for var in self._var_list:
208208
de_var = None

0 commit comments

Comments
 (0)