Skip to content

Commit 4020aa9

Browse files
PWZERrhdong
authored andcommitted
recover _build_eager in saver.restore()
1 parent 7817c0f commit 4020aa9

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow_recommenders_addons/dynamic_embedding/python/ops/tf_save_restore_patch.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,9 @@ def restore(self, sess, save_path):
438438
'TFRADynamicEmbedding')
439439

440440
try:
441-
if not context.executing_eagerly():
441+
if context.executing_eagerly():
442+
self._build_eager(save_path, build_save=False, build_restore=True)
443+
else:
442444
sess.run(self.saver_def.restore_op_name,
443445
{self.saver_def.filename_tensor_name: save_path})
444446
sess.run(self._de_restore_ops,

0 commit comments

Comments
 (0)