config.json

{
    "model_type": "superposition_patch_classifier",
    "embed_dim": 256,
    "patch_dim": 64,
    "n_bootstrap": 2,
    "n_geometric": 2,
    "n_heads": 4,
    "dropout": 0.0,
    "grid_shape": [8, 16, 16],
    "patch_shape": [2, 4, 4],
    "num_patches": 64,
    "local_gate_dim": 11,
    "structural_gate_dim": 6,
    "total_gate_dim": 17,
    "num_classes": 27,
    "num_gates": 5,
    "auto_map": {
        "AutoModel": "geometric_model.SuperpositionPatchClassifier"
    }
}