Hugging Face's logo

braindecode
/
eegpt-pretrained

PyTorch
Safetensors
braindecode
EEGPT
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
 Community
Kkuntal990 commited on
Commit
77a50e6
·
verified ·
1 Parent(s): a2f663c

Push model using huggingface_hub.

Browse files
Files changed (4) hide show
  1. README.md +9 -58
  2. config.json +1286 -6
  3. model.safetensors +2 -2
  4. pytorch_model.bin +3 -0
README.md CHANGED
@@ -1,63 +1,14 @@
1
  ---
2
  library_name: braindecode
3
- tags:
4
- - eeg
5
- - brain-signal-processing
6
- - foundation-model
7
  license: bsd-3-clause
 
 
 
 
 
8
  ---
9
 
10
- # EEGPT Pretrained Model
11
-
12
- Pretrained EEGPT encoder weights for EEG signal processing from [EEGPT: Pretrained Transformer for Universal and Reliable Representation of EEG Signals](https://proceedings.neurips.cc/paper_files/paper/2024/file/4540d267eeec4e5dbd9dae9448f0b739-Paper-Conference.pdf).
13
-
14
- This model was pretrained on a large-scale EEG dataset using masked dual self-supervised learning.
15
-
16
- ## Model Details
17
-
18
- - **Architecture:** Transformer-based encoder with 8 layers
19
- - **Embedding dimension:** 512
20
- - **Number of attention heads:** 8
21
- - **Patch size:** 64
22
- - **Parameters:** ~10M
23
-
24
- ## Usage
25
-
26
- ```python
27
- from braindecode.models import EEGPT
28
- import mne
29
-
30
- # Create channel info (example with 22 channels)
31
- ch_names = ["Fz", "FC3", "FC1", "FCz", "FC2", "FC4", "C5", "C3", "C1", "Cz",
32
- "C2", "C4", "C6", "CP3", "CP1", "CPz", "CP2", "CP4", "P1", "Pz", "P2", "POz"]
33
- ch_types = ["eeg"] * len(ch_names)
34
- info = mne.create_info(ch_names=ch_names, sfreq=250., ch_types=ch_types)
35
-
36
- # Load pretrained model
37
- model = EEGPT.from_pretrained(
38
- "braindecode/eegpt-pretrained",
39
- n_outputs=4, # number of classes for your task
40
- n_chans=22,
41
- n_times=1024,
42
- chs_info=info["chs"],
43
- sfreq=250.0
44
- )
45
-
46
- # Use for downstream tasks
47
- # model can now be fine-tuned on your specific EEG classification task
48
- ```
49
-
50
- ## Citation
51
-
52
- ```bibtex
53
- @inproceedings{tang2024eegpt,
54
- title={EEGPT: Pretrained Transformer for Universal and Reliable Representation of EEG Signals},
55
- author={Tang, Xiwen and others},
56
- booktitle={Advances in Neural Information Processing Systems},
57
- year={2024}
58
- }
59
- ```
60
-
61
- ## License
62
-
63
- BSD-3-Clause (braindecode)
 
1
  ---
2
  library_name: braindecode
 
 
 
 
3
  license: bsd-3-clause
4
+ tags:
5
+ - EEGPT
6
+ - braindecode
7
+ - model_hub_mixin
8
+ - pytorch_model_hub_mixin
9
  ---
10
 
11
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
12
+ - Code: https://braindecode.org
13
+ - Paper: [More Information Needed]
14
+ - Docs: https://braindecode.org/stable/generated/braindecode.models.EEGPT.html
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
config.json CHANGED
@@ -1,14 +1,1294 @@
1
  {
2
- "architectures": [
3
- "EEGPT"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  ],
5
- "model_type": "eegpt",
6
- "embed_dim": 512,
 
7
  "embed_num": 4,
 
8
  "depth": 8,
9
  "num_heads": 8,
10
- "patch_size": 64,
11
- "patch_stride": 32,
12
  "mlp_ratio": 4.0,
13
  "qkv_bias": true,
14
  "drop_prob": 0.0,
 
1
  {
2
+ "n_outputs": 4,
3
+ "n_chans": 58,
4
+ "n_times": 1024,
5
+ "input_window_seconds": null,
6
+ "sfreq": 250.0,
7
+ "chs_info": [
8
+ {
9
+ "ch_name": "FP1",
10
+ "kind": 2,
11
+ "coil_type": 1,
12
+ "unit": 107,
13
+ "cal": 1.0,
14
+ "range": 1.0,
15
+ "loc": [
16
+ NaN,
17
+ NaN,
18
+ NaN,
19
+ NaN,
20
+ NaN,
21
+ NaN,
22
+ NaN,
23
+ NaN,
24
+ NaN,
25
+ NaN,
26
+ NaN,
27
+ NaN
28
+ ]
29
+ },
30
+ {
31
+ "ch_name": "FPZ",
32
+ "kind": 2,
33
+ "coil_type": 1,
34
+ "unit": 107,
35
+ "cal": 1.0,
36
+ "range": 1.0,
37
+ "loc": [
38
+ NaN,
39
+ NaN,
40
+ NaN,
41
+ NaN,
42
+ NaN,
43
+ NaN,
44
+ NaN,
45
+ NaN,
46
+ NaN,
47
+ NaN,
48
+ NaN,
49
+ NaN
50
+ ]
51
+ },
52
+ {
53
+ "ch_name": "FP2",
54
+ "kind": 2,
55
+ "coil_type": 1,
56
+ "unit": 107,
57
+ "cal": 1.0,
58
+ "range": 1.0,
59
+ "loc": [
60
+ NaN,
61
+ NaN,
62
+ NaN,
63
+ NaN,
64
+ NaN,
65
+ NaN,
66
+ NaN,
67
+ NaN,
68
+ NaN,
69
+ NaN,
70
+ NaN,
71
+ NaN
72
+ ]
73
+ },
74
+ {
75
+ "ch_name": "AF7",
76
+ "kind": 2,
77
+ "coil_type": 1,
78
+ "unit": 107,
79
+ "cal": 1.0,
80
+ "range": 1.0,
81
+ "loc": [
82
+ NaN,
83
+ NaN,
84
+ NaN,
85
+ NaN,
86
+ NaN,
87
+ NaN,
88
+ NaN,
89
+ NaN,
90
+ NaN,
91
+ NaN,
92
+ NaN,
93
+ NaN
94
+ ]
95
+ },
96
+ {
97
+ "ch_name": "AF3",
98
+ "kind": 2,
99
+ "coil_type": 1,
100
+ "unit": 107,
101
+ "cal": 1.0,
102
+ "range": 1.0,
103
+ "loc": [
104
+ NaN,
105
+ NaN,
106
+ NaN,
107
+ NaN,
108
+ NaN,
109
+ NaN,
110
+ NaN,
111
+ NaN,
112
+ NaN,
113
+ NaN,
114
+ NaN,
115
+ NaN
116
+ ]
117
+ },
118
+ {
119
+ "ch_name": "AF4",
120
+ "kind": 2,
121
+ "coil_type": 1,
122
+ "unit": 107,
123
+ "cal": 1.0,
124
+ "range": 1.0,
125
+ "loc": [
126
+ NaN,
127
+ NaN,
128
+ NaN,
129
+ NaN,
130
+ NaN,
131
+ NaN,
132
+ NaN,
133
+ NaN,
134
+ NaN,
135
+ NaN,
136
+ NaN,
137
+ NaN
138
+ ]
139
+ },
140
+ {
141
+ "ch_name": "AF8",
142
+ "kind": 2,
143
+ "coil_type": 1,
144
+ "unit": 107,
145
+ "cal": 1.0,
146
+ "range": 1.0,
147
+ "loc": [
148
+ NaN,
149
+ NaN,
150
+ NaN,
151
+ NaN,
152
+ NaN,
153
+ NaN,
154
+ NaN,
155
+ NaN,
156
+ NaN,
157
+ NaN,
158
+ NaN,
159
+ NaN
160
+ ]
161
+ },
162
+ {
163
+ "ch_name": "F7",
164
+ "kind": 2,
165
+ "coil_type": 1,
166
+ "unit": 107,
167
+ "cal": 1.0,
168
+ "range": 1.0,
169
+ "loc": [
170
+ NaN,
171
+ NaN,
172
+ NaN,
173
+ NaN,
174
+ NaN,
175
+ NaN,
176
+ NaN,
177
+ NaN,
178
+ NaN,
179
+ NaN,
180
+ NaN,
181
+ NaN
182
+ ]
183
+ },
184
+ {
185
+ "ch_name": "F5",
186
+ "kind": 2,
187
+ "coil_type": 1,
188
+ "unit": 107,
189
+ "cal": 1.0,
190
+ "range": 1.0,
191
+ "loc": [
192
+ NaN,
193
+ NaN,
194
+ NaN,
195
+ NaN,
196
+ NaN,
197
+ NaN,
198
+ NaN,
199
+ NaN,
200
+ NaN,
201
+ NaN,
202
+ NaN,
203
+ NaN
204
+ ]
205
+ },
206
+ {
207
+ "ch_name": "F3",
208
+ "kind": 2,
209
+ "coil_type": 1,
210
+ "unit": 107,
211
+ "cal": 1.0,
212
+ "range": 1.0,
213
+ "loc": [
214
+ NaN,
215
+ NaN,
216
+ NaN,
217
+ NaN,
218
+ NaN,
219
+ NaN,
220
+ NaN,
221
+ NaN,
222
+ NaN,
223
+ NaN,
224
+ NaN,
225
+ NaN
226
+ ]
227
+ },
228
+ {
229
+ "ch_name": "F1",
230
+ "kind": 2,
231
+ "coil_type": 1,
232
+ "unit": 107,
233
+ "cal": 1.0,
234
+ "range": 1.0,
235
+ "loc": [
236
+ NaN,
237
+ NaN,
238
+ NaN,
239
+ NaN,
240
+ NaN,
241
+ NaN,
242
+ NaN,
243
+ NaN,
244
+ NaN,
245
+ NaN,
246
+ NaN,
247
+ NaN
248
+ ]
249
+ },
250
+ {
251
+ "ch_name": "FZ",
252
+ "kind": 2,
253
+ "coil_type": 1,
254
+ "unit": 107,
255
+ "cal": 1.0,
256
+ "range": 1.0,
257
+ "loc": [
258
+ NaN,
259
+ NaN,
260
+ NaN,
261
+ NaN,
262
+ NaN,
263
+ NaN,
264
+ NaN,
265
+ NaN,
266
+ NaN,
267
+ NaN,
268
+ NaN,
269
+ NaN
270
+ ]
271
+ },
272
+ {
273
+ "ch_name": "F2",
274
+ "kind": 2,
275
+ "coil_type": 1,
276
+ "unit": 107,
277
+ "cal": 1.0,
278
+ "range": 1.0,
279
+ "loc": [
280
+ NaN,
281
+ NaN,
282
+ NaN,
283
+ NaN,
284
+ NaN,
285
+ NaN,
286
+ NaN,
287
+ NaN,
288
+ NaN,
289
+ NaN,
290
+ NaN,
291
+ NaN
292
+ ]
293
+ },
294
+ {
295
+ "ch_name": "F4",
296
+ "kind": 2,
297
+ "coil_type": 1,
298
+ "unit": 107,
299
+ "cal": 1.0,
300
+ "range": 1.0,
301
+ "loc": [
302
+ NaN,
303
+ NaN,
304
+ NaN,
305
+ NaN,
306
+ NaN,
307
+ NaN,
308
+ NaN,
309
+ NaN,
310
+ NaN,
311
+ NaN,
312
+ NaN,
313
+ NaN
314
+ ]
315
+ },
316
+ {
317
+ "ch_name": "F6",
318
+ "kind": 2,
319
+ "coil_type": 1,
320
+ "unit": 107,
321
+ "cal": 1.0,
322
+ "range": 1.0,
323
+ "loc": [
324
+ NaN,
325
+ NaN,
326
+ NaN,
327
+ NaN,
328
+ NaN,
329
+ NaN,
330
+ NaN,
331
+ NaN,
332
+ NaN,
333
+ NaN,
334
+ NaN,
335
+ NaN
336
+ ]
337
+ },
338
+ {
339
+ "ch_name": "F8",
340
+ "kind": 2,
341
+ "coil_type": 1,
342
+ "unit": 107,
343
+ "cal": 1.0,
344
+ "range": 1.0,
345
+ "loc": [
346
+ NaN,
347
+ NaN,
348
+ NaN,
349
+ NaN,
350
+ NaN,
351
+ NaN,
352
+ NaN,
353
+ NaN,
354
+ NaN,
355
+ NaN,
356
+ NaN,
357
+ NaN
358
+ ]
359
+ },
360
+ {
361
+ "ch_name": "FT7",
362
+ "kind": 2,
363
+ "coil_type": 1,
364
+ "unit": 107,
365
+ "cal": 1.0,
366
+ "range": 1.0,
367
+ "loc": [
368
+ NaN,
369
+ NaN,
370
+ NaN,
371
+ NaN,
372
+ NaN,
373
+ NaN,
374
+ NaN,
375
+ NaN,
376
+ NaN,
377
+ NaN,
378
+ NaN,
379
+ NaN
380
+ ]
381
+ },
382
+ {
383
+ "ch_name": "FC5",
384
+ "kind": 2,
385
+ "coil_type": 1,
386
+ "unit": 107,
387
+ "cal": 1.0,
388
+ "range": 1.0,
389
+ "loc": [
390
+ NaN,
391
+ NaN,
392
+ NaN,
393
+ NaN,
394
+ NaN,
395
+ NaN,
396
+ NaN,
397
+ NaN,
398
+ NaN,
399
+ NaN,
400
+ NaN,
401
+ NaN
402
+ ]
403
+ },
404
+ {
405
+ "ch_name": "FC3",
406
+ "kind": 2,
407
+ "coil_type": 1,
408
+ "unit": 107,
409
+ "cal": 1.0,
410
+ "range": 1.0,
411
+ "loc": [
412
+ NaN,
413
+ NaN,
414
+ NaN,
415
+ NaN,
416
+ NaN,
417
+ NaN,
418
+ NaN,
419
+ NaN,
420
+ NaN,
421
+ NaN,
422
+ NaN,
423
+ NaN
424
+ ]
425
+ },
426
+ {
427
+ "ch_name": "FC1",
428
+ "kind": 2,
429
+ "coil_type": 1,
430
+ "unit": 107,
431
+ "cal": 1.0,
432
+ "range": 1.0,
433
+ "loc": [
434
+ NaN,
435
+ NaN,
436
+ NaN,
437
+ NaN,
438
+ NaN,
439
+ NaN,
440
+ NaN,
441
+ NaN,
442
+ NaN,
443
+ NaN,
444
+ NaN,
445
+ NaN
446
+ ]
447
+ },
448
+ {
449
+ "ch_name": "FCZ",
450
+ "kind": 2,
451
+ "coil_type": 1,
452
+ "unit": 107,
453
+ "cal": 1.0,
454
+ "range": 1.0,
455
+ "loc": [
456
+ NaN,
457
+ NaN,
458
+ NaN,
459
+ NaN,
460
+ NaN,
461
+ NaN,
462
+ NaN,
463
+ NaN,
464
+ NaN,
465
+ NaN,
466
+ NaN,
467
+ NaN
468
+ ]
469
+ },
470
+ {
471
+ "ch_name": "FC2",
472
+ "kind": 2,
473
+ "coil_type": 1,
474
+ "unit": 107,
475
+ "cal": 1.0,
476
+ "range": 1.0,
477
+ "loc": [
478
+ NaN,
479
+ NaN,
480
+ NaN,
481
+ NaN,
482
+ NaN,
483
+ NaN,
484
+ NaN,
485
+ NaN,
486
+ NaN,
487
+ NaN,
488
+ NaN,
489
+ NaN
490
+ ]
491
+ },
492
+ {
493
+ "ch_name": "FC4",
494
+ "kind": 2,
495
+ "coil_type": 1,
496
+ "unit": 107,
497
+ "cal": 1.0,
498
+ "range": 1.0,
499
+ "loc": [
500
+ NaN,
501
+ NaN,
502
+ NaN,
503
+ NaN,
504
+ NaN,
505
+ NaN,
506
+ NaN,
507
+ NaN,
508
+ NaN,
509
+ NaN,
510
+ NaN,
511
+ NaN
512
+ ]
513
+ },
514
+ {
515
+ "ch_name": "FC6",
516
+ "kind": 2,
517
+ "coil_type": 1,
518
+ "unit": 107,
519
+ "cal": 1.0,
520
+ "range": 1.0,
521
+ "loc": [
522
+ NaN,
523
+ NaN,
524
+ NaN,
525
+ NaN,
526
+ NaN,
527
+ NaN,
528
+ NaN,
529
+ NaN,
530
+ NaN,
531
+ NaN,
532
+ NaN,
533
+ NaN
534
+ ]
535
+ },
536
+ {
537
+ "ch_name": "FT8",
538
+ "kind": 2,
539
+ "coil_type": 1,
540
+ "unit": 107,
541
+ "cal": 1.0,
542
+ "range": 1.0,
543
+ "loc": [
544
+ NaN,
545
+ NaN,
546
+ NaN,
547
+ NaN,
548
+ NaN,
549
+ NaN,
550
+ NaN,
551
+ NaN,
552
+ NaN,
553
+ NaN,
554
+ NaN,
555
+ NaN
556
+ ]
557
+ },
558
+ {
559
+ "ch_name": "T7",
560
+ "kind": 2,
561
+ "coil_type": 1,
562
+ "unit": 107,
563
+ "cal": 1.0,
564
+ "range": 1.0,
565
+ "loc": [
566
+ NaN,
567
+ NaN,
568
+ NaN,
569
+ NaN,
570
+ NaN,
571
+ NaN,
572
+ NaN,
573
+ NaN,
574
+ NaN,
575
+ NaN,
576
+ NaN,
577
+ NaN
578
+ ]
579
+ },
580
+ {
581
+ "ch_name": "C5",
582
+ "kind": 2,
583
+ "coil_type": 1,
584
+ "unit": 107,
585
+ "cal": 1.0,
586
+ "range": 1.0,
587
+ "loc": [
588
+ NaN,
589
+ NaN,
590
+ NaN,
591
+ NaN,
592
+ NaN,
593
+ NaN,
594
+ NaN,
595
+ NaN,
596
+ NaN,
597
+ NaN,
598
+ NaN,
599
+ NaN
600
+ ]
601
+ },
602
+ {
603
+ "ch_name": "C3",
604
+ "kind": 2,
605
+ "coil_type": 1,
606
+ "unit": 107,
607
+ "cal": 1.0,
608
+ "range": 1.0,
609
+ "loc": [
610
+ NaN,
611
+ NaN,
612
+ NaN,
613
+ NaN,
614
+ NaN,
615
+ NaN,
616
+ NaN,
617
+ NaN,
618
+ NaN,
619
+ NaN,
620
+ NaN,
621
+ NaN
622
+ ]
623
+ },
624
+ {
625
+ "ch_name": "C1",
626
+ "kind": 2,
627
+ "coil_type": 1,
628
+ "unit": 107,
629
+ "cal": 1.0,
630
+ "range": 1.0,
631
+ "loc": [
632
+ NaN,
633
+ NaN,
634
+ NaN,
635
+ NaN,
636
+ NaN,
637
+ NaN,
638
+ NaN,
639
+ NaN,
640
+ NaN,
641
+ NaN,
642
+ NaN,
643
+ NaN
644
+ ]
645
+ },
646
+ {
647
+ "ch_name": "CZ",
648
+ "kind": 2,
649
+ "coil_type": 1,
650
+ "unit": 107,
651
+ "cal": 1.0,
652
+ "range": 1.0,
653
+ "loc": [
654
+ NaN,
655
+ NaN,
656
+ NaN,
657
+ NaN,
658
+ NaN,
659
+ NaN,
660
+ NaN,
661
+ NaN,
662
+ NaN,
663
+ NaN,
664
+ NaN,
665
+ NaN
666
+ ]
667
+ },
668
+ {
669
+ "ch_name": "C2",
670
+ "kind": 2,
671
+ "coil_type": 1,
672
+ "unit": 107,
673
+ "cal": 1.0,
674
+ "range": 1.0,
675
+ "loc": [
676
+ NaN,
677
+ NaN,
678
+ NaN,
679
+ NaN,
680
+ NaN,
681
+ NaN,
682
+ NaN,
683
+ NaN,
684
+ NaN,
685
+ NaN,
686
+ NaN,
687
+ NaN
688
+ ]
689
+ },
690
+ {
691
+ "ch_name": "C4",
692
+ "kind": 2,
693
+ "coil_type": 1,
694
+ "unit": 107,
695
+ "cal": 1.0,
696
+ "range": 1.0,
697
+ "loc": [
698
+ NaN,
699
+ NaN,
700
+ NaN,
701
+ NaN,
702
+ NaN,
703
+ NaN,
704
+ NaN,
705
+ NaN,
706
+ NaN,
707
+ NaN,
708
+ NaN,
709
+ NaN
710
+ ]
711
+ },
712
+ {
713
+ "ch_name": "C6",
714
+ "kind": 2,
715
+ "coil_type": 1,
716
+ "unit": 107,
717
+ "cal": 1.0,
718
+ "range": 1.0,
719
+ "loc": [
720
+ NaN,
721
+ NaN,
722
+ NaN,
723
+ NaN,
724
+ NaN,
725
+ NaN,
726
+ NaN,
727
+ NaN,
728
+ NaN,
729
+ NaN,
730
+ NaN,
731
+ NaN
732
+ ]
733
+ },
734
+ {
735
+ "ch_name": "T8",
736
+ "kind": 2,
737
+ "coil_type": 1,
738
+ "unit": 107,
739
+ "cal": 1.0,
740
+ "range": 1.0,
741
+ "loc": [
742
+ NaN,
743
+ NaN,
744
+ NaN,
745
+ NaN,
746
+ NaN,
747
+ NaN,
748
+ NaN,
749
+ NaN,
750
+ NaN,
751
+ NaN,
752
+ NaN,
753
+ NaN
754
+ ]
755
+ },
756
+ {
757
+ "ch_name": "TP7",
758
+ "kind": 2,
759
+ "coil_type": 1,
760
+ "unit": 107,
761
+ "cal": 1.0,
762
+ "range": 1.0,
763
+ "loc": [
764
+ NaN,
765
+ NaN,
766
+ NaN,
767
+ NaN,
768
+ NaN,
769
+ NaN,
770
+ NaN,
771
+ NaN,
772
+ NaN,
773
+ NaN,
774
+ NaN,
775
+ NaN
776
+ ]
777
+ },
778
+ {
779
+ "ch_name": "CP5",
780
+ "kind": 2,
781
+ "coil_type": 1,
782
+ "unit": 107,
783
+ "cal": 1.0,
784
+ "range": 1.0,
785
+ "loc": [
786
+ NaN,
787
+ NaN,
788
+ NaN,
789
+ NaN,
790
+ NaN,
791
+ NaN,
792
+ NaN,
793
+ NaN,
794
+ NaN,
795
+ NaN,
796
+ NaN,
797
+ NaN
798
+ ]
799
+ },
800
+ {
801
+ "ch_name": "CP3",
802
+ "kind": 2,
803
+ "coil_type": 1,
804
+ "unit": 107,
805
+ "cal": 1.0,
806
+ "range": 1.0,
807
+ "loc": [
808
+ NaN,
809
+ NaN,
810
+ NaN,
811
+ NaN,
812
+ NaN,
813
+ NaN,
814
+ NaN,
815
+ NaN,
816
+ NaN,
817
+ NaN,
818
+ NaN,
819
+ NaN
820
+ ]
821
+ },
822
+ {
823
+ "ch_name": "CP1",
824
+ "kind": 2,
825
+ "coil_type": 1,
826
+ "unit": 107,
827
+ "cal": 1.0,
828
+ "range": 1.0,
829
+ "loc": [
830
+ NaN,
831
+ NaN,
832
+ NaN,
833
+ NaN,
834
+ NaN,
835
+ NaN,
836
+ NaN,
837
+ NaN,
838
+ NaN,
839
+ NaN,
840
+ NaN,
841
+ NaN
842
+ ]
843
+ },
844
+ {
845
+ "ch_name": "CPZ",
846
+ "kind": 2,
847
+ "coil_type": 1,
848
+ "unit": 107,
849
+ "cal": 1.0,
850
+ "range": 1.0,
851
+ "loc": [
852
+ NaN,
853
+ NaN,
854
+ NaN,
855
+ NaN,
856
+ NaN,
857
+ NaN,
858
+ NaN,
859
+ NaN,
860
+ NaN,
861
+ NaN,
862
+ NaN,
863
+ NaN
864
+ ]
865
+ },
866
+ {
867
+ "ch_name": "CP2",
868
+ "kind": 2,
869
+ "coil_type": 1,
870
+ "unit": 107,
871
+ "cal": 1.0,
872
+ "range": 1.0,
873
+ "loc": [
874
+ NaN,
875
+ NaN,
876
+ NaN,
877
+ NaN,
878
+ NaN,
879
+ NaN,
880
+ NaN,
881
+ NaN,
882
+ NaN,
883
+ NaN,
884
+ NaN,
885
+ NaN
886
+ ]
887
+ },
888
+ {
889
+ "ch_name": "CP4",
890
+ "kind": 2,
891
+ "coil_type": 1,
892
+ "unit": 107,
893
+ "cal": 1.0,
894
+ "range": 1.0,
895
+ "loc": [
896
+ NaN,
897
+ NaN,
898
+ NaN,
899
+ NaN,
900
+ NaN,
901
+ NaN,
902
+ NaN,
903
+ NaN,
904
+ NaN,
905
+ NaN,
906
+ NaN,
907
+ NaN
908
+ ]
909
+ },
910
+ {
911
+ "ch_name": "CP6",
912
+ "kind": 2,
913
+ "coil_type": 1,
914
+ "unit": 107,
915
+ "cal": 1.0,
916
+ "range": 1.0,
917
+ "loc": [
918
+ NaN,
919
+ NaN,
920
+ NaN,
921
+ NaN,
922
+ NaN,
923
+ NaN,
924
+ NaN,
925
+ NaN,
926
+ NaN,
927
+ NaN,
928
+ NaN,
929
+ NaN
930
+ ]
931
+ },
932
+ {
933
+ "ch_name": "TP8",
934
+ "kind": 2,
935
+ "coil_type": 1,
936
+ "unit": 107,
937
+ "cal": 1.0,
938
+ "range": 1.0,
939
+ "loc": [
940
+ NaN,
941
+ NaN,
942
+ NaN,
943
+ NaN,
944
+ NaN,
945
+ NaN,
946
+ NaN,
947
+ NaN,
948
+ NaN,
949
+ NaN,
950
+ NaN,
951
+ NaN
952
+ ]
953
+ },
954
+ {
955
+ "ch_name": "P7",
956
+ "kind": 2,
957
+ "coil_type": 1,
958
+ "unit": 107,
959
+ "cal": 1.0,
960
+ "range": 1.0,
961
+ "loc": [
962
+ NaN,
963
+ NaN,
964
+ NaN,
965
+ NaN,
966
+ NaN,
967
+ NaN,
968
+ NaN,
969
+ NaN,
970
+ NaN,
971
+ NaN,
972
+ NaN,
973
+ NaN
974
+ ]
975
+ },
976
+ {
977
+ "ch_name": "P5",
978
+ "kind": 2,
979
+ "coil_type": 1,
980
+ "unit": 107,
981
+ "cal": 1.0,
982
+ "range": 1.0,
983
+ "loc": [
984
+ NaN,
985
+ NaN,
986
+ NaN,
987
+ NaN,
988
+ NaN,
989
+ NaN,
990
+ NaN,
991
+ NaN,
992
+ NaN,
993
+ NaN,
994
+ NaN,
995
+ NaN
996
+ ]
997
+ },
998
+ {
999
+ "ch_name": "P3",
1000
+ "kind": 2,
1001
+ "coil_type": 1,
1002
+ "unit": 107,
1003
+ "cal": 1.0,
1004
+ "range": 1.0,
1005
+ "loc": [
1006
+ NaN,
1007
+ NaN,
1008
+ NaN,
1009
+ NaN,
1010
+ NaN,
1011
+ NaN,
1012
+ NaN,
1013
+ NaN,
1014
+ NaN,
1015
+ NaN,
1016
+ NaN,
1017
+ NaN
1018
+ ]
1019
+ },
1020
+ {
1021
+ "ch_name": "P1",
1022
+ "kind": 2,
1023
+ "coil_type": 1,
1024
+ "unit": 107,
1025
+ "cal": 1.0,
1026
+ "range": 1.0,
1027
+ "loc": [
1028
+ NaN,
1029
+ NaN,
1030
+ NaN,
1031
+ NaN,
1032
+ NaN,
1033
+ NaN,
1034
+ NaN,
1035
+ NaN,
1036
+ NaN,
1037
+ NaN,
1038
+ NaN,
1039
+ NaN
1040
+ ]
1041
+ },
1042
+ {
1043
+ "ch_name": "PZ",
1044
+ "kind": 2,
1045
+ "coil_type": 1,
1046
+ "unit": 107,
1047
+ "cal": 1.0,
1048
+ "range": 1.0,
1049
+ "loc": [
1050
+ NaN,
1051
+ NaN,
1052
+ NaN,
1053
+ NaN,
1054
+ NaN,
1055
+ NaN,
1056
+ NaN,
1057
+ NaN,
1058
+ NaN,
1059
+ NaN,
1060
+ NaN,
1061
+ NaN
1062
+ ]
1063
+ },
1064
+ {
1065
+ "ch_name": "P2",
1066
+ "kind": 2,
1067
+ "coil_type": 1,
1068
+ "unit": 107,
1069
+ "cal": 1.0,
1070
+ "range": 1.0,
1071
+ "loc": [
1072
+ NaN,
1073
+ NaN,
1074
+ NaN,
1075
+ NaN,
1076
+ NaN,
1077
+ NaN,
1078
+ NaN,
1079
+ NaN,
1080
+ NaN,
1081
+ NaN,
1082
+ NaN,
1083
+ NaN
1084
+ ]
1085
+ },
1086
+ {
1087
+ "ch_name": "P4",
1088
+ "kind": 2,
1089
+ "coil_type": 1,
1090
+ "unit": 107,
1091
+ "cal": 1.0,
1092
+ "range": 1.0,
1093
+ "loc": [
1094
+ NaN,
1095
+ NaN,
1096
+ NaN,
1097
+ NaN,
1098
+ NaN,
1099
+ NaN,
1100
+ NaN,
1101
+ NaN,
1102
+ NaN,
1103
+ NaN,
1104
+ NaN,
1105
+ NaN
1106
+ ]
1107
+ },
1108
+ {
1109
+ "ch_name": "P6",
1110
+ "kind": 2,
1111
+ "coil_type": 1,
1112
+ "unit": 107,
1113
+ "cal": 1.0,
1114
+ "range": 1.0,
1115
+ "loc": [
1116
+ NaN,
1117
+ NaN,
1118
+ NaN,
1119
+ NaN,
1120
+ NaN,
1121
+ NaN,
1122
+ NaN,
1123
+ NaN,
1124
+ NaN,
1125
+ NaN,
1126
+ NaN,
1127
+ NaN
1128
+ ]
1129
+ },
1130
+ {
1131
+ "ch_name": "P8",
1132
+ "kind": 2,
1133
+ "coil_type": 1,
1134
+ "unit": 107,
1135
+ "cal": 1.0,
1136
+ "range": 1.0,
1137
+ "loc": [
1138
+ NaN,
1139
+ NaN,
1140
+ NaN,
1141
+ NaN,
1142
+ NaN,
1143
+ NaN,
1144
+ NaN,
1145
+ NaN,
1146
+ NaN,
1147
+ NaN,
1148
+ NaN,
1149
+ NaN
1150
+ ]
1151
+ },
1152
+ {
1153
+ "ch_name": "PO7",
1154
+ "kind": 2,
1155
+ "coil_type": 1,
1156
+ "unit": 107,
1157
+ "cal": 1.0,
1158
+ "range": 1.0,
1159
+ "loc": [
1160
+ NaN,
1161
+ NaN,
1162
+ NaN,
1163
+ NaN,
1164
+ NaN,
1165
+ NaN,
1166
+ NaN,
1167
+ NaN,
1168
+ NaN,
1169
+ NaN,
1170
+ NaN,
1171
+ NaN
1172
+ ]
1173
+ },
1174
+ {
1175
+ "ch_name": "PO5",
1176
+ "kind": 2,
1177
+ "coil_type": 1,
1178
+ "unit": 107,
1179
+ "cal": 1.0,
1180
+ "range": 1.0,
1181
+ "loc": [
1182
+ NaN,
1183
+ NaN,
1184
+ NaN,
1185
+ NaN,
1186
+ NaN,
1187
+ NaN,
1188
+ NaN,
1189
+ NaN,
1190
+ NaN,
1191
+ NaN,
1192
+ NaN,
1193
+ NaN
1194
+ ]
1195
+ },
1196
+ {
1197
+ "ch_name": "PO3",
1198
+ "kind": 2,
1199
+ "coil_type": 1,
1200
+ "unit": 107,
1201
+ "cal": 1.0,
1202
+ "range": 1.0,
1203
+ "loc": [
1204
+ NaN,
1205
+ NaN,
1206
+ NaN,
1207
+ NaN,
1208
+ NaN,
1209
+ NaN,
1210
+ NaN,
1211
+ NaN,
1212
+ NaN,
1213
+ NaN,
1214
+ NaN,
1215
+ NaN
1216
+ ]
1217
+ },
1218
+ {
1219
+ "ch_name": "POZ",
1220
+ "kind": 2,
1221
+ "coil_type": 1,
1222
+ "unit": 107,
1223
+ "cal": 1.0,
1224
+ "range": 1.0,
1225
+ "loc": [
1226
+ NaN,
1227
+ NaN,
1228
+ NaN,
1229
+ NaN,
1230
+ NaN,
1231
+ NaN,
1232
+ NaN,
1233
+ NaN,
1234
+ NaN,
1235
+ NaN,
1236
+ NaN,
1237
+ NaN
1238
+ ]
1239
+ },
1240
+ {
1241
+ "ch_name": "PO4",
1242
+ "kind": 2,
1243
+ "coil_type": 1,
1244
+ "unit": 107,
1245
+ "cal": 1.0,
1246
+ "range": 1.0,
1247
+ "loc": [
1248
+ NaN,
1249
+ NaN,
1250
+ NaN,
1251
+ NaN,
1252
+ NaN,
1253
+ NaN,
1254
+ NaN,
1255
+ NaN,
1256
+ NaN,
1257
+ NaN,
1258
+ NaN,
1259
+ NaN
1260
+ ]
1261
+ },
1262
+ {
1263
+ "ch_name": "PO8",
1264
+ "kind": 2,
1265
+ "coil_type": 1,
1266
+ "unit": 107,
1267
+ "cal": 1.0,
1268
+ "range": 1.0,
1269
+ "loc": [
1270
+ NaN,
1271
+ NaN,
1272
+ NaN,
1273
+ NaN,
1274
+ NaN,
1275
+ NaN,
1276
+ NaN,
1277
+ NaN,
1278
+ NaN,
1279
+ NaN,
1280
+ NaN,
1281
+ NaN
1282
+ ]
1283
+ }
1284
  ],
1285
+ "braindecode_version": "1.3.1",
1286
+ "patch_size": 64,
1287
+ "patch_stride": 32,
1288
  "embed_num": 4,
1289
+ "embed_dim": 512,
1290
  "depth": 8,
1291
  "num_heads": 8,
 
 
1292
  "mlp_ratio": 4.0,
1293
  "qkv_bias": true,
1294
  "drop_prob": 0.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ddff4d9a60f8afb7647537bc586c9b97fe18f4955b92ffb123b2dc868e4e7908
3
- size 101157984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ab6f5f347ef7d77ad0ba8a75dc4058ef9fa7f6d548e7383b30d125ddeca7342
3
+ size 102175504
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c54a8345dfdabbc34875f1a2a0e11f5395733d9bc83573548d6122a9f51bd68
3
+ size 102204099