Hugging Face's logo

braindecode
/
codebrain-pretrained

PyTorch
Safetensors
braindecode
CodeBrain
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
 Community
codebrain-pretrained
File size: 1,820 Bytes
1ba4589
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
{
  "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"
}