brant-pretrained / config.json
adammnrr's picture
Add Brant pretrained weights (paper config, 508M params)
ee386ce verified
Raw
History Blame Contribute Delete
387 Bytes
{
"patch_size": 1500,
"embed_dim": 2048,
"ffn_dim": 3072,
"temporal_n_layers": 12,
"spatial_n_layers": 5,
"n_heads": 16,
"n_freq_bands": 8,
"drop_prob": 0.1,
"activation": "torch.nn.modules.activation.ReLU",
"n_outputs": 2,
"n_chans": 16,
"chs_info": null,
"n_times": 22500,
"input_window_seconds": null,
"sfreq": 250,
"braindecode_version": "1.7.0dev0"
}