1

我有以下问题主要是由于我缺乏对 TF 的专业知识，特别是 TF2.0。

我尝试将 Elmo 嵌入与 keras LSTM 实现一起使用。

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

这就是我所做的：

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

来自 Tf-Hub 的 Elmo 嵌入类

  • 现在，我创建了一个 tf.function，因为我已经读到它是 TF2.0 方式等效于 Session.run()。尽管 autograph 参数默认设置为 TRUE，但对我来说主要问题是如何将图形传递给 Elmo TF-Hub 模块......？

Elmo LSTM tf.function

最后一次调用产生以下错误：

运行时错误

4

1 回答 1

2

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

否则你可能会调查这个：

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

于 2020-10-19T12:36:01.017 回答