你好~我想在Java中使用BERT嵌入。当我将模型转换为.pb格式时，我可以获得嵌入，但是句子的最大长度是128,而我希望修改句子的最大长度为30。你能告诉我如何解决这个问题吗？
ie3xauqp1#
你好，如何在Java中加载带有.pb文件的BERT?谢谢。
nbnkbykc2#
我尝试使用.pb文件，但是无法更改句子的最大长度。
5us2dqdw3#
我找到了https://github.com/robrua/easy-bert。这个代码更清晰。我认为@256785,你不能改变它，这是你的模型在一开始就训练过的。
shyt4zoc4#
是的，我也读过这个。也许我无法改变它。但是当我使用提取句子特征的Python脚本时，max_length参数可以改变。所以我猜我忽略了一些东西。此外，当我使用easy_bert时，花费的时间太长了。
4条答案
按热度按时间ie3xauqp1#
你好，如何在Java中加载带有.pb文件的BERT?谢谢。
nbnkbykc2#
我尝试使用.pb文件，但是无法更改句子的最大长度。
5us2dqdw3#
我找到了https://github.com/robrua/easy-bert。这个代码更清晰。我认为@256785,你不能改变它，这是你的模型在一开始就训练过的。
shyt4zoc4#
是的，我也读过这个。也许我无法改变它。但是当我使用提取句子特征的Python脚本时，max_length参数可以改变。所以我猜我忽略了一些东西。此外，当我使用easy_bert时，花费的时间太长了。