config.json

{
  "activation": "gelu",
  "dropout": 0.1,
  "embed_dim": 64,
  "hex_conv_layers": 2,
  "hex_hidden_dim": 32,
  "in_channels": 13,
  "lattice_type": "hexagonal",
  "num_output_tokens": 4,
  "offset_type": "even-q",
  "spatial_size": 8,
  "use_batch_norm": true,
  "use_crystalline": true
}