英文

sentence-transformers/gtr-t5-xl

这是一个 sentence-transformers 模型:它将句子和段落映射到一个768维的稠密向量空间。该模型特别用于语义搜索任务。

该模型是从TensorFlow模型 gtr-xl-1 转换为PyTorch模型。使用此模型时,请参阅发布的文章: Large Dual Encoders Are Generalizable Retrievers 。tfhub模型和PyTorch模型可能会产生略有不同的嵌入,但在相同的基准测试上运行时,它们会产生相同的结果。

该模型仅使用T5-3B模型的编码器。权重以FP16格式存储。

使用(Sentence-Transformers)

如果已经安装了 sentence-transformers ,则使用此模型变得很容易:

pip install -U sentence-transformers

然后,您可以像这样使用该模型:

from sentence_transformers import SentenceTransformer
sentences = ["This is an example sentence", "Each sentence is converted"]

model = SentenceTransformer('sentence-transformers/gtr-t5-xl')
embeddings = model.encode(sentences)
print(embeddings)

该模型要求使用sentence-transformers版本2.2.0或更新版本。

评估结果

有关对该模型进行自动评估的结果,请参阅句子嵌入基准(Sentence Embeddings Benchmark): https://seb.sbert.net

引用和作者

如果您发现该模型有帮助,请引用相应的文章: Large Dual Encoders Are Generalizable Retrievers