{
  "run_id": "snake-balanced-multilingual-0923-005225",
  "base_model": "hub:aac6fef/laya-multilingual-mlx",
  "base_model_dir": "/Users/biplovgautam/.cache/huggingface/hub/models--aac6fef--laya-multilingual-mlx/snapshots/f2b4faf51023039425946074e2cf1361d2db11d5",
  "dataset": "snake-moves-generated-on-this-mac-970a6be8",
  "dataset_sha256": "970a6be8b1b458204b95fa747c66b4c527512bdf373f549f88d97391243cd24a",
  "hyperparameters": {
    "method": "lora",
    "objective": "proper",
    "epochs": 4,
    "batch_size": 8,
    "grad_accum": 2,
    "lr": 0.0002,
    "head_lr": 0.0001,
    "lora_rank": 16,
    "lora_alpha": 32,
    "lora_dropout": 0.05,
    "lora_layers": 0,
    "full_layers": 4,
    "head_dropout": 0.1,
    "weight_decay": 0.01,
    "warmup": 0.06,
    "max_grad_norm": 1.0,
    "shuffle_options": true,
    "class_weighting": "none",
    "patience": 2,
    "grad_checkpoint": "auto",
    "precision": "bfloat16",
    "seed": 13
  },
  "trainable_params": 18150145,
  "total_params": 325288198,
  "train_decisions": 2339,
  "val_decisions": 261,
  "updates": 588,
  "best_epoch": 3,
  "best_val_loss": 0.05348039334880479,
  "history": [
    {
      "epoch": 1,
      "val_loss": 0.05746363440483888,
      "val_accuracy": 0.9923371647509579
    },
    {
      "epoch": 2,
      "val_loss": 0.07788544589805121,
      "val_accuracy": 0.9808429118773946
    },
    {
      "epoch": 3,
      "val_loss": 0.05348039334880479,
      "val_accuracy": 0.9923371647509579
    },
    {
      "epoch": 4,
      "val_loss": 0.06205711615945649,
      "val_accuracy": 0.9923371647509579
    }
  ],
  "train_seconds": 1100.8,
  "peak_memory_gb": 3.52,
  "gradient_checkpointing": false,
  "calibration": {
    "ece_uncalibrated": 0.005269409375311282,
    "ece_calibrated": 0.0034790613387207436,
    "temperature": [
      1.3116,
      1.0,
      1.0
    ],
    "temperature_by_options": {
      "choice:3-5": 1.3116
    },
    "fitted_types": [
      "choice"
    ]
  },
  "created": "2026-09-23T01:11:23"
}