Add ST-EEGFormer base (ported from ST-EEGFormer-base)

README.md

@@ -0,0 +1,49 @@
+---
+library_name: braindecode
+tags:
+- braindecode
+- STEEGFormer
+- eeg
+- foundation-model
+license: mit
+---
+
+# STEEGFormer (base)
+
+ViT-MAE EEG foundation model — braindecode port of **ST-EEGFormer** (base variant).
+
+## Provenance
+
+- **Weights ported from:** [LiuyinYang1101/STEEGFormer](https://github.com/LiuyinYang1101/STEEGFormer),
+  release [`ST-EEGFormer-base`](https://github.com/LiuyinYang1101/STEEGFormer/releases/tag/ST-EEGFormer-base)
+  (asset `checkpoint-288.pth`).
+- **Upstream license:** MIT. The braindecode wrapper code is BSD-3-Clause.
+- The pre-trained encoder is loaded faithfully (numerical equivalence verified:
+  pre-encoder bit-exact, post-encoder relative error ~4e-6). The MAE decoder is
+  dropped and the classification head is re-initialised.
+
+## Architecture
+
+| | embed_dim | depth | num_heads | patch_size | channel vocab |
+|---|---|---|---|---|---|
+| base | 768 | 12 | 12 | 16 | 145 |
+
+Channel positions are resolved from electrode names in `chs_info` (145-slot shared montage vocabulary).
+
+## Usage
+
+```python
+from braindecode.models import STEEGFormer
+
+model = STEEGFormer.from_pretrained(
+    "braindecode/STEEGFormer-base",
+    n_outputs=4, n_chans=22, n_times=1000, chs_info=chs_info,
+)
+# Encoder features:  out = model(x, return_features=True); out["features"]
+```
+
+## Citation
+
+Yang, L., Sun, Q., Li, A. & Van Hulle, M. M. (2026). *Are EEG foundation models
+worth it? Comparative evaluation with traditional decoders in diverse BCI tasks.*
+ICLR 2026. https://openreview.net/forum?id=5Xwm8e6vbh

config.json

@@ -0,0 +1,20 @@
+{
+  "n_outputs": 4,
+  "n_chans": 22,
+  "chs_info": null,
+  "n_times": 1000,
+  "input_window_seconds": null,
+  "sfreq": null,
+  "patch_size": 16,
+  "embed_dim": 768,
+  "depth": 12,
+  "num_heads": 12,
+  "mlp_ratio": 4.0,
+  "drop_prob": 0.0,
+  "drop_path": 0.0,
+  "activation": "torch.nn.modules.activation.GELU",
+  "global_pool": "avg",
+  "n_chans_pos": 145,
+  "chan_pos_idx": null,
+  "braindecode_version": "1.6.1dev0"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9fe864b700544a1a832da258ec64525f888dc80be790c977ee0d669b6bc652a
+size 340756448

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6be3bbe0edc2cd47d7030475010e5fdbb4b49c88bf9478148c4a6f875a6599c3
+size 340811523