赞
踩
我使用bert进行了文本分类,想使用保存的模型对一些新的数据进行预测
发现会遇到错误ValueError: Unknown layer: TokenEmbedding。
万般搜索不得其解,在网上看到了一个方案https://blog.csdn.net/yscoder/article/details/103214078,尝试之后成功。