如何用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'未定义＇我如何加载我保存到路径的模型?

Answer 1

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

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