{
  "model_type": "kona_sudoku_ebm",
  "architecture": "KonaEnergyModel",
  "in_channels": 10,
  "hidden_dim": 64,
  "num_heads": 4,
  "num_layers": 2,
  "mlp_hidden": 128,
  "dropout": 0.25,
  "checkpoint_file": "kona_ebm_best.pt",
  "training_data": "data/train_large.txt",
  "best_epoch": 9,
  "framework": "pytorch",
  "task": "sudoku-solving"
}