codebrain-pretrained / config.json
bruAristimunha's picture
Add CodeBrain pretrained weights (TUEG, 19ch 10-20, 200Hz)
1ba4589 verified
{
"n_outputs": 2,
"n_chans": 19,
"chs_info": [
{
"ch_name": "C3",
"kind": "eeg"
},
{
"ch_name": "C4",
"kind": "eeg"
},
{
"ch_name": "Cz",
"kind": "eeg"
},
{
"ch_name": "F3",
"kind": "eeg"
},
{
"ch_name": "F4",
"kind": "eeg"
},
{
"ch_name": "Fp1",
"kind": "eeg"
},
{
"ch_name": "Fp2",
"kind": "eeg"
},
{
"ch_name": "F7",
"kind": "eeg"
},
{
"ch_name": "F8",
"kind": "eeg"
},
{
"ch_name": "Fz",
"kind": "eeg"
},
{
"ch_name": "O1",
"kind": "eeg"
},
{
"ch_name": "O2",
"kind": "eeg"
},
{
"ch_name": "P3",
"kind": "eeg"
},
{
"ch_name": "P4",
"kind": "eeg"
},
{
"ch_name": "Pz",
"kind": "eeg"
},
{
"ch_name": "T3",
"kind": "eeg"
},
{
"ch_name": "T4",
"kind": "eeg"
},
{
"ch_name": "T5",
"kind": "eeg"
},
{
"ch_name": "T6",
"kind": "eeg"
}
],
"n_times": 6000,
"input_window_seconds": 30.0,
"sfreq": 200.0,
"patch_size": 200,
"res_channels": 200,
"skip_channels": 200,
"out_channels": 200,
"num_res_layers": 8,
"drop_prob": 0.1,
"s4_bidirectional": true,
"s4_layernorm": false,
"s4_lmax": 570,
"s4_d_state": 64,
"conv_out_chans": 25,
"conv_groups": 5,
"proj_kernel_size": 49,
"proj_padding": 24,
"proj_refine_kernel": 3,
"pos_kernel": [
19,
7
],
"spectral_dropout": 0.1,
"mlp_hidden_multiplier": 4,
"swa_window_size": 1,
"codebook_size_t": 4096,
"codebook_size_f": 4096,
"pretrain_mode": false,
"activation": "torch.nn.modules.activation.ReLU",
"braindecode_version": "1.3.2"
}