加载预训练通用句子编码器的问题

Question

我在为 NLP 任务加载预训练模块时遇到问题，问题是由于我想的 tf 迁移。 Tensorflow 网站说，如果正确给出签名变量，问题可能会得到解决。 你能帮我更正这段代码吗？

类型错误：'AutoTrackable' object 不可调用

[代码]

import tensorflow_hub as hub
# enabling the pretrained model for trainig our custom model using tensorflow hub
module_url = "https://tfhub.dev/google/universal-sentence-encoder-large/3"
embed = hub.load(module_url)

# creating a method for embedding and will using method for every input layer 
def UniversalEmbedding(x):
    return embed(tf.squeeze(tf.cast(x, tf.string)), signature='default', as_dict=True)["default"]

Answer 1

我是这个领域的新手，但我可以与您分享我为解决问题所做的工作：

• 使用以下命令检查您的 Tensorflow 版本：

  打印（张量流。版本）

• 您可以在运行 import tensorflow 之前运行以下代码来更改 Tensorflow 版本

%tensorflow_version 1.x

• 我的 TF 版本是 1.x，我正在使用hub.load(url)并收到“Auto Trackable”错误消息，当我用hub.Module(url)替换它时，它解决了我的问题

• 您可能会发现这些链接很有帮助！ 如果你不知道他们

https://tfhub.dev/google/elmo/3 https://www.tensorflow.org/hub/common_issues

祝你好运