{
  "model": {
    "width": 32,
    "blocks": [
      2,
      2,
      2,
      2
    ],
    "num_color": 7,
    "num_type": 7,
    "dropout": 0.2
  },
  "labels": {
    "color": [
      "White",
      "Blue",
      "Black",
      "Red",
      "Green",
      "Colorless",
      "Multicolor"
    ],
    "type": [
      "Creature",
      "Planeswalker",
      "Land",
      "Artifact",
      "Enchantment",
      "Instant",
      "Sorcery"
    ]
  },
  "normalization": {
    "mean": [
      0.412272,
      0.37507,
      0.34918
    ],
    "std": [
      0.251613,
      0.230892,
      0.231858
    ]
  },
  "image_size": [
    160,
    224
  ],
  "dials": {
    "run_name": "light-aug",
    "epochs": 30,
    "batch_size": 256,
    "lr": 0.002,
    "weight_decay": 0.05,
    "warmup_frac": 0.05,
    "width": 32,
    "blocks": [
      2,
      2,
      2,
      2
    ],
    "dropout": 0.2,
    "loss": "ce",
    "label_smoothing": 0.1,
    "focal_gamma": 2.0,
    "type_loss_weight": 1.0,
    "color_loss_weight": 1.0,
    "img_size": [
      160,
      224
    ],
    "eval_every_epochs": 1,
    "patience": 6,
    "grad_clip": 1.0,
    "hflip": false,
    "aug": "light",
    "limit": null,
    "seed": 42
  },
  "params": 2800366,
  "dtype": "bfloat16",
  "train_state": {
    "epoch": 20,
    "best_epoch": 20,
    "best_score": 0.3503234434514614,
    "best_val": {
      "val_loss": 2.9443500779889558,
      "val_color_acc": 0.3781546537452049,
      "val_color_macro_f1": 0.368955477464954,
      "val_type_acc": 0.5659196446598022,
      "val_type_macro_f1": 0.3316914094379687,
      "val_score": 0.3503234434514614
    }
  },
  "source_checkpoint": "/home/nathank/BITS/dev/MTG Card Classifier/checkpoints/light-aug/best",
  "exported_at": "2026-10-23 20:04:46",
  "metrics_test": {
    "split": "test",
    "n": 3401,
    "evaluated_at": "2026-10-23 20:04:39",
    "color": {
      "accuracy": 0.3684210526315789,
      "top2_accuracy": 0.5698324022346368,
      "macro_f1": 0.3450303818057278,
      "ece": 0.04191294569227494
    },
    "type": {
      "accuracy": 0.5292561011467215,
      "top2_accuracy": 0.6745074977947663,
      "macro_f1": 0.2856992909416024,
      "ece": 0.05380913312909744
    }
  },
  "latency": {
    "samples": 100,
    "gpu_bf16": {
      "n": 100,
      "mean_s": 0.00373,
      "p50_s": 0.00368,
      "p95_s": 0.00424,
      "max_s": 0.00475,
      "ratio_to_benchmark": 0.05,
      "device": "NVIDIA GeForce RTX 3090 Ti"
    },
    "cpu_fp32": {
      "n": 100,
      "mean_s": 0.0052,
      "p50_s": 0.00505,
      "p95_s": 0.00601,
      "max_s": 0.00809,
      "ratio_to_benchmark": 0.07,
      "threads": 16
    },
    "benchmark_s": 0.074
  },
  "model_py_sha1": "8d93607d00c223dba206254a9a03c1e24f23bd97"
}