training_metadata.json

{
  "base_model": "sentence-transformers/all-MiniLM-L6-v2",
  "training_corpus_size": 63474,
  "training_examples": 63474,
  "train_examples": 57126,
  "validation_examples": 6348,
  "validation_split": 0.1,
  "chunk_size": 256,
  "chunk_overlap": 50,
  "batch_size": 128,
  "epochs": 5,
  "learning_rate": 2e-05,
  "warmup_ratio": 0.1,
  "early_stopping_patience": 5,
  "eval_steps": 200,
  "device": "cuda",
  "training_api": "SentenceTransformers_v3+",
  "loss_function": "MultipleNegativesRankingLoss",
  "eval_strategy": "steps",
  "fp16": true,
  "save_strategy": "steps",
  "save_steps": 200,
  "logging_steps": 50,
  "save_total_limit": 3,
  "push_to_hub": true,
  "hub_model_id": "RikoteMaster/retriever_pdf_and_books",
  "hub_private": false
}