{
  "base": "BAAI/bge-base-en-v1.5",
  "pairs": "data/pairs_augmented",
  "n_pairs": 135868,
  "epochs": 2,
  "max_steps": 1800,
  "scheduler": "cosine",
  "warmup_ratio": 0.05,
  "checkpoint_metric": "eval_dev_graded_ndcg@10",
  "batch": 256,
  "lr": 2e-05,
  "max_seq": 128,
  "dev_distractors": 50000,
  "matryoshka_dims": [
    768,
    384,
    128,
    64
  ],
  "loss": "MatryoshkaLoss(CachedMultipleNegativesRankingLoss)",
  "mini_batch_size": 16,
  "query_prompt": "Represent this sentence for searching relevant passages: ",
  "batch_sampler": "NO_DUPLICATES",
  "precision": "bf16",
  "device": "cuda"
}