Sentence Transformer

Overview

This is the tensorflow implementation of [Sentence Transformer](https://www.sbert.net/). As of now `tf-transformers` support following models. .. code-block:

'sentence-transformers/gtr-t5-base',
'sentence-transformers/gtr-t5-large',
'sentence-transformers/gtr-t5-xl',
'sentence-transformers/sentence-t5-base',
'sentence-transformers/sentence-t5-large',
'sentence-transformers/sentence-t5-xl',
'sentence-transformers/msmarco-distilbert-dot-v5',
'sentence-transformers/stsb-distilbert-base',
'sentence-transformers/distilbert-multilingual-nli-stsb-quora-ranking',
'sentence-transformers/msmarco-distilbert-base-v4',
'sentence-transformers/msmarco-distilbert-cos-v5',
'sentence-transformers/msmarco-distilbert-base-v2',
'sentence-transformers/distilbert-base-nli-mean-tokens',
'sentence-transformers/multi-qa-distilbert-cos-v1',
'sentence-transformers/all-MiniLM-L6-v2',
'sentence-transformers/all-MiniLM-L6-v1',
'sentence-transformers/all-MiniLM-L12-v1',
'sentence-transformers/all-MiniLM-L12-v2',
'sentence-transformers/multi-qa-MiniLM-L6-dot-v1',
'sentence-transformers/multi-qa-MiniLM-L6-cos-v1',
'sentence-transformers/paraphrase-MiniLM-L3-v2',
'sentence-transformers/paraphrase-MiniLM-L6-v2',
'sentence-transformers/paraphrase-MiniLM-L12-v2',
'sentence-transformers/msmarco-MiniLM-L6-cos-v5',
'sentence-transformers/msmarco-MiniLM-L12-cos-v5',
'sentence-transformers/all-distilroberta-v1'

Usage:

>>> from tf_transformers.models import SentenceTransformer
>>> model = SentenceTransformer.from_pretrained('sentence-transformers/msmarco-MiniLM-L6-cos-v5')

All model checkpoints can be saved using model.save_checkpoint. For saving checkpoints:

SentenceTransformer