| { |
| "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 |
| } |
| } |
|
|
