config.json

{
  "n_outputs": 5,
  "n_chans": 4,
  "chs_info": null,
  "n_times": 3840,
  "input_window_seconds": null,
  "sfreq": 128.0,
  "patch_size": 640,
  "embed_dim": 128,
  "staging_num_heads": 4,
  "staging_num_layers": 1,
  "staging_pooling_heads": 4,
  "drop_prob": 0.3,
  "max_seq_length": 8196,
  "activation": "torch.nn.modules.activation.ELU"
}