{
  "run_id": "prompt-injection-balanced-english-0922-233445",
  "base_model": "hub:aac6fef/laya-mlx",
  "base_model_dir": "/Users/biplovgautam/.cache/huggingface/hub/models--aac6fef--laya-mlx/snapshots/20aed815fc6acde75733882e7ec0e3f28aeb9717",
  "dataset": "prompt-injection-yes-no-c29679e6",
  "dataset_sha256": "c29679e690b93a9862bbae757a7429ad8d60fd323b7c59a324ad83cc301b41f3",
  "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": 33444865,
  "total_params": 428490502,
  "train_decisions": 492,
  "val_decisions": 54,
  "updates": 124,
  "best_epoch": 4,
  "best_val_loss": 0.15077615171355052,
  "history": [
    {
      "epoch": 1,
      "val_loss": 0.6401203491736576,
      "val_accuracy": 0.8703703703703703
    },
    {
      "epoch": 2,
      "val_loss": 0.4105575770939093,
      "val_accuracy": 0.9074074074074074
    },
    {
      "epoch": 3,
      "val_loss": 0.1956661067346657,
      "val_accuracy": 0.9444444444444444
    },
    {
      "epoch": 4,
      "val_loss": 0.15077615171355052,
      "val_accuracy": 0.9629629629629629
    }
  ],
  "train_seconds": 339.6,
  "peak_memory_gb": 12.34,
  "gradient_checkpointing": true,
  "calibration": {
    "ece_uncalibrated": 0.026573144472652063,
    "ece_calibrated": 0.0138072612947517,
    "temperature": [
      1.6369030475616455,
      1.2514300346374512,
      1.5395
    ],
    "temperature_by_options": {
      "choice:3-5": 1.7601518630981445,
      "choice:6-10": 1.0000158548355103,
      "score:3-5": 1.2514300346374512,
      "choice:11+": 0.10058280825614929,
      "choice:2": 1.9063563346862793,
      "noul:2": 1.5395
    },
    "fitted_types": [
      "noul"
    ]
  },
  "created": "2026-09-22T23:40:37"
}