Невозможно загрузить точно настроенную модель BERT - PullRequest
0 голосов
/ 02 марта 2020

Я использую Tensorflow версия: 1.15.0, Керас версия: 2.2.4, python версия: 3.6

во время обучения я использовал две строки сверху:

import os
os.environ['TF_KERAS'] = '1'

Я сохранил модель, используя следующий код:

from tensorflow.keras.callbacks import EarlyStopping,ModelCheckpoint

es = EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=5)
mc = ModelCheckpoint("model_rento_18.h5", monitor='val_acc', mode='max', verbose=1, save_best_only=True)
model.fit([ sub_seq_train ,sub_mask_train, text_seq_train ,text_mask_train ],Labels_train,epochs=epochs, validation_data=([sub_seq_val,sub_mask_val,text_seq_val,text_mask_val],Labels_val),batch_size = batch_size,callbacks=[es, mc] )

Теперь я пытаюсь загрузить модель в той же среде, используя код ниже:

import os
os.environ['TF_KERAS'] = '1'
from tensorflow.keras.models import load_model
model = load_model('model.h5')

Я получаю следующую ошибку:

TypeError: init () получил неожиданный аргумент ключевого слова 'ragged'

Пожалуйста, дайте мне знать, если я делаю что-то неправильно или какая-либо версия не соответствует? Заранее спасибо.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...