{
  "d_model": 1024,
  "hidden_dim": 1024,
  "input_dim": 768,
  "metadata": {
    "base_model": "jinaai/jina-embeddings-v5-text-nano",
    "batch_size": 16,
    "dataset": "stsb",
    "dataset_limit": 5749,
    "epochs": 5,
    "init_head_dir": null,
    "init_metadata": {},
    "input_dim": 768,
    "lambda_preserve": 0.1,
    "lambda_snap": 0.05,
    "last_metrics": {
      "cosine_loss": 0.01358454953879118,
      "loss": 0.0290202833712101,
      "preserve_loss": 0.028068706393241882,
      "snap_loss": 0.2525772750377655
    },
    "lr": 0.0002,
    "model_task": "text-matching",
    "output_dim": 1024,
    "steps": 1800
  },
  "output_dim": 1024
}