Instructions to use dmrau/crossencoder-msmarco with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dmrau/crossencoder-msmarco with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="dmrau/crossencoder-msmarco")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("dmrau/crossencoder-msmarco") model = AutoModelForSequenceClassification.from_pretrained("dmrau/crossencoder-msmarco") - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This model was trained with the Rankers Library please check https://github.com/davidmrau/rankers#Evaluation to learn how to use the model.
This Cross Encoder Model was trained on the official MS MARCO training triples using the following training parameters:
batch size: 128
learning rate 3e-6
Optimizer: Adam
Performance TREC DL Nist test 2020:
- NDCG@10: 69.35
- Downloads last month
- 3