如何加载微调的BERT模型

时间:2020-06-18 11:06:17

标签: keras bert-language-model

我已经对数据进行了微调,并使用js script

保存了该模型

现在正尝试使用以下内容加载

model.save()

但我一直收到以下错误

from keras_radam import RAdam
from keras.models import load_model
from keras_bert import get_custom_objects

custom_object = get_custom_objects()
custom_object['RAdam'] = RAdam()

model = load_model('bert_20news.h5', custom_objects=custom_object)

0 个答案:

没有答案