bert4keras icon indicating copy to clipboard operation
bert4keras copied to clipboard

Serving model by bert-as-service.

Open Jhangsy opened this issue 4 years ago • 2 comments

提问时请尽可能提供如下信息:

基本信息

问题描述

使用 bert4keras 训练了一个分类模型输出了 save_weights 输出了 ckpt,model.load_weights 预测正常。 但是用 bert-as-service 结果不正确,观察了输出的 prob,每一个都很小,类似于 1/num_class 的值。 可能与 bert4keras 无关,不知道有没有同学遇到这个问题。 感谢帮助!

Jhangsy avatar Jul 16 '21 07:07 Jhangsy

model.save_weights保存下来的并不是原版bert的格式,原则上用bert-as-service是没法加载的。

bojone avatar Jul 16 '21 08:07 bojone

model.save_weights保存下来的并不是原版bert的格式,原则上用bert-as-service是没法加载的。

谢谢苏神,bert-as-service 可能处于巧合成功加载了,那有没有什么方式可以保存成原版 bert 的格式。

Jhangsy avatar Jul 16 '21 08:07 Jhangsy