| { | |
| "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" | |
| } | |