我得到了这个问题与埃尔莫和tensorflow,我想修复它没有降级。我该怎么办
`**CODE**
import tensorflow_hub as hub
import tensorflow as tf
#Elmo
elmo = hub.Module("https://tfhub.dev/google/elmo/2", trainable=True)
# Provide input tensor and create embeddings
input_tensor = ["my birthday is the best day of the year"]
embeddings_tensor = elmo(input_tensor, signature="default", as_dict=True)["elmo"]
* here i got the problem *
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
embeddings = sess.run(embeddings_tensor)
print(embeddings.shape)
print(embeddings)
`
1条答案
按热度按时间vfh0ocws1#
hub.module()
只与TF 1.x
兼容,不建议在TF 2.x
中使用。我们可以在TF 2.x
代码中使用hub.load()
来代替它。您可以通过在
TF 2.8
(或TF 2.x)中进行以下更改来运行代码而不会出错。输出量:
或者,可以使用以下代码在您的代码上方运行相同的代码(通过将其转换为
TF 1.x
),它将成功执行: