STEEGFormer-small / config.json
bruAristimunha's picture
Add ST-EEGFormer small (ported from ST-EEGFormer-small)
6ebec2d verified
Raw
History Blame Contribute Delete
417 Bytes
{
"n_outputs": 4,
"n_chans": 22,
"chs_info": null,
"n_times": 1000,
"input_window_seconds": null,
"sfreq": null,
"patch_size": 16,
"embed_dim": 512,
"depth": 8,
"num_heads": 8,
"mlp_ratio": 4.0,
"drop_prob": 0.0,
"drop_path": 0.0,
"activation": "torch.nn.modules.activation.GELU",
"global_pool": "avg",
"n_chans_pos": 145,
"chan_pos_idx": null,
"braindecode_version": "1.6.1dev0"
}