是否有一个Tensorflow函数来初始化所有表[文本分类]



我正在处理Tensorflow中的一个推理模型问题。我已经从SavedModel中加载了一个自定义对象(特定度量(的字典。我已经建立了一个推理模型,并给它一个原始字符串的张量,但我收到了错误:

FailedPreconditionError: Exception encountered when calling layer "string_lookup" (type StringLookup).
Table not initialized. [Op:LookupTableFindV2]

因此,我终于用这个不推荐使用的方法(tensorflow 2.7(初始化了表:

tf.compat.v1.initialize_all_tables(name='init_all_tables')

正如你想象的那样,它不起作用。

所以我在subjet link1 link2 link3 上找到了那些票

我的问题是:在加载模型进行推理建模之前,是否需要使用遗留init选项的特定命令来保存模型?

我的模型没有在其中集成vectorize_layer的adapt步骤。有必要将其外部化。

因此,为了恢复我的模型并用它进行推理,解决方案只是重新加载我的模型,并在一瞬间重新适配同一列车文本数据上的vector_layer。

最新更新