Skip to content

Commit 7817c0f

Browse files
PWZERrhdong
authored andcommitted
recover _build_eager in saver.save()
1 parent 2e59f91 commit 7817c0f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/tf_save_restore_patch.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,12 @@ def save(self,
354354

355355
if not self._is_empty:
356356
try:
357-
if not context.executing_eagerly():
357+
if context.executing_eagerly():
358+
self._build_eager(checkpoint_file,
359+
build_save=True,
360+
build_restore=False)
361+
model_checkpoint_path = self.saver_def.save_tensor_name
362+
else:
358363
model_checkpoint_path = sess.run(
359364
self.saver_def.save_tensor_name,
360365
{self.saver_def.filename_tensor_name: checkpoint_file})

0 commit comments

Comments
 (0)