Data: official LRAT training pairs only; no external training data
Records per epoch: 96,504
Cumulative epochs: 2
Optimizer steps per epoch: 12,063
Hardware: 2 × NVIDIA A40
Precision: bfloat16
Train group: 1 positive and 5 negatives
Global effective query batch: 8
Query / passage maximum length: 128 / 512
Learning rate: 1e-6 for each epoch
Pooling: last token
Normalized embeddings: yes
Temperature: 0.02
Query instruction: Given a web search query, retrieve relevant passages that answer the query

The second epoch starts from the first epoch's full model weights and uses a freshly initialized optimizer and learning-rate schedule. Training uses the official LRAT pair data and reweight_rate values with cross-device negatives. The repository contains complete model parameters and does not require a LoRA or adapter at inference time.

Loading

python

import torch
import torch.nn.functional as F
from transformers import AutoModel, AutoTokenizer

model_id = "Seraphic663/lrat-qwen3-0.6b-2epoch-20260717"
tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side="left")
model = AutoModel.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
).cuda().eval()

instruction = "Given a web search query, retrieve relevant passages that answer the query"
texts = [f"Instruct: {instruction}\nQuery:example query", "example passage"]
batch = tokenizer(
    texts,
    padding=True,
    truncation=True,
    max_length=512,
    return_tensors="pt",
).to("cuda")

with torch.inference_mode():
    output = model(**batch).last_hidden_state
    embeddings = F.normalize(output[:, -1].float(), p=2, dim=1)

Validation Note

On a fixed 500-row candidate-set training-overlap diagnostic, this checkpoint obtained Recall@1/5/10 of 0.738/0.942/0.990 and MRR of 0.82813. All 500 queries also occur in the training pairs, so these values are regression diagnostics, not held-out results or official BC-Plus leaderboard scores.

Integrity

text

model.safetensors
2,383,139,480 bytes
SHA-256 1075292002c27eee654f83e1d1d568a78d4cc9238bc48a36d4ceb5f35968e050