brainbert-pretrained / config.json
adammnrr's picture
Add BrainBERT 'large' pretrained weights (ported from czlwang/BrainBERT)
670f9de verified
Raw
History Blame Contribute Delete
386 Bytes
{
"hidden_dim": 768,
"ffn_dim": 3072,
"n_layers": 6,
"n_heads": 12,
"nperseg": 400,
"noverlap": 350,
"freq_cutoff": 40,
"stft_clip": 5,
"activation": "torch.nn.modules.activation.GELU",
"drop_prob": 0.1,
"n_outputs": 2,
"n_chans": 1,
"chs_info": null,
"n_times": 2048,
"input_window_seconds": null,
"sfreq": 2048.0,
"braindecode_version": "1.7.0dev0"
}