LSTM-来自TF-Hub的Elmo，采用TF2.0完全实现

时间：2020-01-09 23:24:52

标签： lstm tensorflow2.0 elmo tf-hub

我之所以遇到以下问题，主要是因为我缺乏对TF特别是TF2.0的专业知识。

我尝试将Elmo嵌入与keras LSTM实现结合使用。

受启发的代码是使用TF-Hub Elmo模块。如果我想编写100％TF2.0 Python代码，这似乎很难使用。

这就是我所做的：

  • 我已经创建了以下ElmoEmbedding类，但是已经在渴望模式的Colab上遇到了问题。我可以禁用它还是应该将hub.Module调用作为TF2.0迁移文档中建议的默认Graph包括在内？

Elmo Embedding class from Tf-Hub

  • 现在，我已经创建了一个tf.function，因为我已经知道它是与Session.run（）等效的TF2.0方法。对我来说，主要问题是，尽管autograph参数默认情况下设置为TRUE，如何将图形传递给Elmo TF-Hub模块...？

Elmo LSTM tf.function

上次呼叫产生以下错误：

Runtime error

欢迎任何帮助：）

最诚挚的问候

Jerome

1 个答案:

答案 0 :(得分：0)

我遇到了同样的问题，只是决定使用tensorflow 1.15来兼容tensorflow-hub。

否则，您可能会对此进行调查：

https://www.tensorflow.org/hub/migration_tf2

相关问题
最新问题