config.json

{
  "bde_dim": 4,
  "depth": 3,
  "dropout": 0.1,
  "embed_dim": 128,
  "head_dim": 32,
  "heads": 4,
  "mlp_hidden_dim": 128,
  "num_classes": 3,
  "num_electrodes": 62
}