我使用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
如何成功保存和加载模型?