configs/axomamba.json

{
  "num_input": 1278,
  "num_output": 2,
  "num_branch": 118,
  "num_synapse_per_branch": 11,
  "input_to_synapse_routing": "neuronio_routing",
  "model_dim": 58,
  "num_layers": 2,
  "block_repeats": 1,
  "state_dim": 23,
  "conv_kernel": 4,
  "expansion": 2,
  "block_type": "mamba1",
  "dropout": 0.0,
  "residual_scale_init": 0.1,
  "synapse_gain_scale": null,
  "morphology_synapse_gain_scale": 1.0,
  "morphology_synapse_feature_dim": 9,
  "morphology_synapse_feature_hidden": 16,
  "morphology_synapse_feature_scale": 1.0,
  "branch_gain": false,
  "branch_nonlinearity": "tanh",
  "branch_nonlinearity_scale_init": 0.1,
  "branch_bilinear_rank": 0,
  "branch_bilinear_scale": null,
  "separate_heads": true,
  "soma_multiplicative_gate": false,
  "soma_multiplicative_scale_init": 0.1,
  "spike_voltage_coupling_scale": 4.0,
  "integrative_mixer": false,
  "integrative_mixer_short_kernel": 9,
  "integrative_mixer_long_kernel": 65,
  "integrative_mixer_position": "pre",
  "integrative_mixer_long_path_bias": 0.5,
  "local_tcn_scale": 1.0,
  "local_tcn_mode": "dilated",
  "local_tcn_position": "post",
  "morphology_ids": [
    "specimen_479770916",
    "specimen_487664663",
    "specimen_488679042",
    "specimen_488683425",
    "specimen_589427435"
  ],
  "morphology_embedding_scale": 0.15
}