stage1_metadata.json

{
  "step": 9000,
  "checkpoint_file": "step_9000.pt",
  "timestamp": "2026-03-12T22:48:49.369182+00:00",
  "cka_history_keys": [
    "0_0",
    "1_1",
    "3_2",
    "4_3",
    "6_4"
  ],
  "final_loss": 0.012602645438164473,
  "best_loss": 0.011982748401351273,
  "training_config": {
    "loss_type": "mse+cosine",
    "cka_threshold": 0.75,
    "seed": 42
  },
  "stage": 1,
  "stage_name": "layer-alignment"
}