Я попытался следовать ответу по этой ссылке для сохранения модели.
Сохранение и выполнение вывода с использованием модели Tensorflow BERT
Хотя я могу сохранить и загрузить назад модель. Я не могу делать прогнозы, используя загруженную модель. Код, который я использовал для загрузки модели, приведен ниже.
from tensorflow.contrib import predictor
predict_fn = predictor.from_saved_model("./savedModel/1580195293") #the exported model dir
Я не уверен насчет формата, который необходимо использовать для прогнозов, т. Е. Формата, в котором Foregnate_fn должен получать аргументы. , или если код, который я использую, неверен.
Может кто-нибудь помочь с этим.
Пример обзора IMDB mov ie, который похож на мой вариант использования . (Для справки) https://colab.research.google.com/github/google-research/bert/blob/master/predicting_movie_reviews_with_bert_on_tf_hub.ipynb#scrollTo = ERkTE8-7oQLZ