rajpurkar/squad
Viewer • Updated • 98.2k • 212k • 462
How to use jmvcoelho/t5-base-msmarco-squad-query-generation-longp-v2 with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("jmvcoelho/t5-base-msmarco-squad-query-generation-longp-v2")
model = AutoModelForSeq2SeqLM.from_pretrained("jmvcoelho/t5-base-msmarco-squad-query-generation-longp-v2", device_map="auto")t5-base-msmarco-squad-query-generation-longp-v2
Task: query generation Architecture: LongT5
Base model: google/long-t5-tglobal-base
Note: This is supposed to be a baseline model.
"Generate Query: {document}. Query:"
1536 tokens
Batch size: 8;
Gradient acc: 8;
LR: 3e-4, linear scheduler, 400 warmup steps.
Total: 252059 pairs (document, query)
From MARCO-V2: 165238 From SQuAD: 86821
The remaining queries from MARCO-V2 train split were not used.
This model is supposed to be used for data augmentation. Hence, meaningful evaluation will come from downstream tasks.
MARCO-V2 Dev1: BLEU: 0.102 ROUGE: 0.447
MARCO-V2 Dev2: BLEU: 0.1691 ROUGE: 0.5013