WarmStartHook bug: AttributeError: 'Tensor' object has no attribute '_resource_handle'

[lixiang-repo]

System information

• OS Platform and Distribution : Ubuntu 20.04.4
• TensorFlow version and how it was installed (source or binary): 2.8.3 and binary
• TensorFlow-Recommenders-Addons version and how it was installed (source or binary): v0.6.0 binary
• Python version: 3.8.10
• Is GPU used? (yes/no): no

Describe the bug

WarmStartHook has a bug 修改代码：tensorflow_recommenders_addons/dynamic_embedding/python/ops/warm_ start_util.py]. 把saveable.op打印出来，报错日志和改动如下：

Code to reproduce the issue 复现代码和报错日志，以及复现bug流程: 训练一次，保存checkout，目录是ckpt，然后embedding会保存成csv文件，第一列是id， 后面是32列向量， mv ckpt ckpt1, 修改warm_path=ckpt1， 重新执行run.sh训练脚本,就会出现下面报错日志

log.error.txt

code.tar.gz

[MoFHeka]

Has the bug been fixed?

[MoFHeka]

@Mr-Nineteen Could you solve this problem if it is convenient for you?