{
  "run_id": "prompt-injection-demo-run-0923-021926",
  "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": 2,
    "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": 62,
  "best_epoch": 2,
  "best_val_loss": 0.26240006766942303,
  "history": [
    {
      "epoch": 1,
      "val_loss": 0.32080752853523864,
      "val_accuracy": 0.9074074074074074
    },
    {
      "epoch": 2,
      "val_loss": 0.26240006766942303,
      "val_accuracy": 0.9444444444444444
    }
  ],
  "train_seconds": 245.0,
  "peak_memory_gb": 3.36,
  "gradient_checkpointing": true,
  "calibration": {
    "ece_uncalibrated": 0.04960620039834296,
    "ece_calibrated": 0.028031846416004265,
    "temperature": [
      1.6369030475616455,
      1.2514300346374512,
      1.7648
    ],
    "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.7648
    },
    "fitted_types": [
      "noul"
    ]
  },
  "created": "2026-09-23T02:23:48"
}