在R中保存和加载BERT模型

时间:2020-06-29 13:49:16

标签: r keras bert-language-model

我使用this tutorial在某些文本数据上部分训练了模型(预训练的BERT)。除了数据,其他所有代码都是相同的。

我在保存和加载经过训练的模型时遇到问题。如果我使用以下代码:

save_model_hdf5(model, "model.h5")
model2 <- load_model_hdf5("model.h5")

我收到此错误:

Error in py_call_impl(callable, dots$args, dots$keywords) : 
  ValueError: Unknown layer: TokenEmbedding 

如何成功保存和加载模型?

0 个答案:

没有答案