nlp - 如何从本地路径加载带有 SentenceTransformers 的 Bert 预训练模型？

Question

我正在使用 SentenceTransformer 库来使用 Bert 预训练模型我在 google Colabs 中下载了文件并使用以下命令保存了它：

from sentence_transformers import SentenceTransformer
model = SentenceTransformer('bert-large-uncased-whole-word-masking')
model.save('/content/drive/MyDrive/Pizza/Bert_Model')

这些给了我一个包含我的模型的文件夹，然后我下载该文件夹并想在我的本地文件中使用以下代码：

from sentence_transformers import SentenceTransformer
model = SentenceTransformer('Bert_Model')

但现在它不能正常工作，我得到这个错误，' NameError: name 'model' is not defined ' 如何加载我保存到路径的模型？

Answer 1

当您从本地文件加载模型时，您必须指定模型的路径，而不是其名称。这意味着 的参数SentenceTransformer()必须是包含文件的config.json文件夹的完整路径：

from sentence_transformers import SentenceTransformer
model = SentenceTransformer('/path/to/your/model')