LSTM - Elmo от TF-Hub с чистой реализацией TF2.0 - PullRequest
0 голосов
/ 10 января 2020

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

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

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

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

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

Elmo Embedding class from Tf-Hub

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

Elmo LSTM tf.function

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

Runtime error

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

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

Джером

...