Сохранить модель в керасе

Question

Извините за мой наивный вопрос, но я пытаюсь сохранить свою модель keras (), в которой я использую функцию TFBertModel () в качестве скрытого слоя. Для этого я использую функцию save (), предоставляемую пакетом tf.keras.

Но я получил эту ошибку:

---------------------------------------------------------------------------

NotImplementedError                       Traceback (most recent call last)

<ipython-input-13-3b315f7219da> in <module>()
----> 1 model.save('model_weights.h5')

8 frames

/tensorflow-2.1.0/python3.6/tensorflow_core/python/keras/engine/network.py in get_config(self)
    915   def get_config(self):
    916     if not self._is_graph_network:
--> 917       raise NotImplementedError
    918     return copy.deepcopy(get_network_config(self))
    919 

NotImplementedError: 

Ошибка может быть воспроизведена из моего колба: https://colab.research.google.com/drive/18HYwffkXCylPqeA-8raL82vfwOjb-aLP

И еще один вопрос: как мне назвать эту модель для предсказания?

Спасибо за вашу помощь!

Answer 1

Не могу ответить вам, не увидев код, но вам, вероятно, стоит посмотреть здесь:

https://towardsdatascience.com/text-classification-with-hugging-face-transformers-in-tensorflow-2-without-tears-ee50e4f3e7ed