microsoft/ms_marco
Viewer • Updated • 1.11M • 24.5k • 247
How to use lx865712528/master-base-pretrained-msmarco with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("feature-extraction", model="lx865712528/master-base-pretrained-msmarco") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("lx865712528/master-base-pretrained-msmarco")
model = AutoModelForMaskedLM.from_pretrained("lx865712528/master-base-pretrained-msmarco", device_map="auto")Paper: https://arxiv.org/abs/2212.07841.
Code: https://github.com/microsoft/SimXNS/tree/main/MASTER.
This is the checkpoint after pretraining on the MS-MARCO corpus. You may use this checkpoint as the initialization for finetuning.
To load this checkpoint for initialization, you may follow:
from transformers import AutoModel
model = AutoModel.from_pretrained('lx865712528/master-base-pretrained-msmarco')