LSTM - Elmo от TF-Hub с чистой реализацией TF2.0

Question

У меня есть следующая проблема, в основном из-за моего отсутствия опыта работы с TF в целом и TF2.0 в частности.

Я пытаюсь использовать встраивание Elmo с реализацией keras LSTM.

Код, на который я вдохновлен, использует модуль TF-Hub Elmo. Что, по-видимому, довольно сложно использовать, если я хочу написать 100% -ный код Python TF2.0.

Вот что я сделал:

• Я создал следующий класс ElmoEmbedding, но уже столкнулся с проблемой Colab в нетерпеливом режиме. Могу ли я отключить его или включить вызов hub. Module в Graph по умолчанию, как это предлагается в документации по миграции TF2.0?

• Теперь я создал tf.function, поскольку прочитал, что это способ TF2.0, эквивалентный Session.run(). Главный вопрос для меня, несмотря на то, что для параметра autograph по умолчанию установлено значение TRUE, заключается в том, как график будет передан в модуль Elmo TF-Hub...?

Последний вызов вызывает следующую ошибку:

Любая помощь приветствуется:)

С наилучшими пожеланиями

Джером

Answer 1

У меня была такая же проблема, и я просто решил использовать вместо нее tenorflow 1.15, который совместим с tenorflow-hub.

В противном случае вы могли бы изучить это:

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