duin-pretrained / config.json
adammnrr's picture
Du-IN pretrained weights (24 subject x stage checkpoints) converted for braindecode.models.DuIN (PR #1298)
7409e2a verified
Raw History Blame Contribute Delete
594 Bytes
{
"n_outputs": 61,
"n_chans": 10,
"chs_info": null,
"n_times": 3000,
"input_window_seconds": null,
"sfreq": 1000.0,
"patch_size": 100,
"n_neural": 16,
"n_filters": [
128,
128,
16
],
"kernel_sizes": [
19,
3,
3
],
"strides": [
10,
1,
1
],
"n_layers": 8,
"n_heads": 8,
"head_dim": 64,
"ffn_dim": 320,
"head_hidden_dim": 128,
"activation": "torch.nn.modules.activation.ReLU",
"attn_drop_prob": 0.2,
"drop_prob": 0.2,
"channel_strategy": "native",
"channel_strategy_kwargs": null,
"braindecode_version": "1.8.1"
}