{
  "schema_version": 1,
  "passed": true,
  "step": 76030,
  "epochs_completed": 5,
  "train_items_total": 15206,
  "train_items_used_per_epoch": 15206,
  "train_items_with_positive_labels": 12988,
  "train_zero_positive_items_used_per_epoch": 2218,
  "steps_per_epoch": 15206,
  "val_items_total": 3802,
  "val_items_with_positive_labels": 3281,
  "best_val_loss": 0.8114464077540036,
  "best_epoch": 5,
  "stop_reason": "max_epochs_completed",
  "early_stopping": {
    "patience": 5,
    "min_delta": 0.0,
    "bad_epochs": 1,
    "best_epoch": 4
  },
  "final_val_metrics": {
    "loss": 0.8401362389553415,
    "recall_at_5": 0.7849824111608121,
    "evaluated_items": 3802.0,
    "evaluated_positive_items": 3281.0,
    "evaluated_zero_positive_items": 521.0
  },
  "final_model_path": "outputs/models/context_sphere_v3_final.pt",
  "best_model_path": "outputs/models/context_sphere_v3_best.pt",
  "history_tail": [
    {
      "step": 76012,
      "epoch": 5,
      "loss": 1.3384813070297241,
      "grad_norm": 4.108997344970703
    },
    {
      "step": 76013,
      "epoch": 5,
      "loss": 1.2154099941253662,
      "grad_norm": 2.668102264404297
    },
    {
      "step": 76014,
      "epoch": 5,
      "loss": 1.2979867458343506,
      "grad_norm": 4.211376190185547
    },
    {
      "step": 76015,
      "epoch": 5,
      "loss": 1.1624202728271484,
      "grad_norm": 1.0014667510986328
    },
    {
      "step": 76016,
      "epoch": 5,
      "loss": 1.0331001281738281,
      "grad_norm": 1.7403678894042969
    },
    {
      "step": 76017,
      "epoch": 5,
      "loss": 1.2406458854675293,
      "grad_norm": 4.224254131317139
    },
    {
      "step": 76018,
      "epoch": 5,
      "loss": 1.2704728841781616,
      "grad_norm": 0.8251820802688599
    },
    {
      "step": 76019,
      "epoch": 5,
      "loss": 1.0506269931793213,
      "grad_norm": 1.216317892074585
    },
    {
      "step": 76020,
      "epoch": 5,
      "loss": 1.2293930053710938,
      "grad_norm": 1.5133010149002075
    },
    {
      "step": 76021,
      "epoch": 5,
      "loss": 2.0957672595977783,
      "grad_norm": 19.02604866027832
    },
    {
      "step": 76022,
      "epoch": 5,
      "loss": 1.321169137954712,
      "grad_norm": 0.7611741423606873
    },
    {
      "step": 76023,
      "epoch": 5,
      "loss": 0.8465029001235962,
      "grad_norm": 3.9663000106811523
    },
    {
      "step": 76024,
      "epoch": 5,
      "loss": 0.9299266934394836,
      "grad_norm": 1.597723126411438
    },
    {
      "step": 76025,
      "epoch": 5,
      "loss": 1.1610701084136963,
      "grad_norm": 1.9924956560134888
    },
    {
      "step": 76026,
      "epoch": 5,
      "loss": 0.9163417816162109,
      "grad_norm": 1.8209384679794312
    },
    {
      "step": 76027,
      "epoch": 5,
      "loss": 1.337764859199524,
      "grad_norm": 2.7863895893096924
    },
    {
      "step": 76028,
      "epoch": 5,
      "loss": 0.7450914978981018,
      "grad_norm": 1.4205079078674316
    },
    {
      "step": 76029,
      "epoch": 5,
      "loss": 1.1089109182357788,
      "grad_norm": 3.8144006729125977
    },
    {
      "step": 76030,
      "epoch": 5,
      "loss": 1.0382972955703735,
      "grad_norm": 2.1582326889038086
    },
    {
      "step": 76030,
      "epoch": 5,
      "train_loss": 1.19866059164785,
      "val_loss": 0.8401362389553415,
      "val_recall_at_5": 0.7849824111608121,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 1,
      "is_best_epoch": false
    }
  ],
  "epoch_summaries": [
    {
      "epoch": 1,
      "step": 15206,
      "train_loss": 1.2529381703824005,
      "val_loss": 0.8538370929160662,
      "val_recall_at_5": 0.7725897856059061,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 0,
      "is_best_epoch": true
    },
    {
      "epoch": 2,
      "step": 30412,
      "train_loss": 1.2103027913518036,
      "val_loss": 0.8377963499073479,
      "val_recall_at_5": 0.777610541790602,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 0,
      "is_best_epoch": true
    },
    {
      "epoch": 3,
      "step": 45618,
      "train_loss": 1.2033953751472521,
      "val_loss": 0.8403141896217388,
      "val_recall_at_5": 0.7825559769898318,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 1,
      "is_best_epoch": false
    },
    {
      "epoch": 4,
      "step": 60824,
      "train_loss": 1.2007089401724527,
      "val_loss": 0.837044046802749,
      "val_recall_at_5": 0.7843819158266226,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 0,
      "is_best_epoch": true
    },
    {
      "epoch": 5,
      "step": 76030,
      "train_loss": 1.19866059164785,
      "val_loss": 0.8401362389553415,
      "val_recall_at_5": 0.7849824111608121,
      "val_evaluated_items": 3802.0,
      "val_evaluated_positive_items": 3281.0,
      "val_evaluated_zero_positive_items": 521.0,
      "early_stopping_bad_epochs": 1,
      "is_best_epoch": false
    }
  ],
  "config": {
    "train_count": 15206,
    "val_count": 3802,
    "raw_train_count": 15206,
    "raw_val_count": 3802,
    "train_positive_count": 12988,
    "val_positive_count": 3281,
    "train_positive_only": false,
    "steps_per_epoch": 15206,
    "val_max_items": 0,
    "device": "cuda",
    "epochs": 5,
    "learning_rate": 0.001,
    "weight_decay": 0.0001,
    "positive_weight": 4.0,
    "pair_dim": 32,
    "tile_size": 32,
    "gradient_checkpointing": true,
    "early_stopping_patience": 5,
    "early_stopping_min_delta": 0.0
  }
}