无法加载经过微调的BERT模型

时间:2020-03-02 11:32:36

标签: python-3.x tensorflow keras pre-trained-model

我正在使用 Tensorflow 版本:1.15.0, Keras 版本: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 ()得到了意外的关键字参数“参差不齐”

如果我做任何不正确或任何版本不匹配的事情,请告诉我? 预先感谢。

0 个答案:

没有答案