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

README.md

@@ -0,0 +1,49 @@
+---
+library_name: braindecode
+tags:
+- braindecode
+- STEEGFormer
+- eeg
+- foundation-model
+license: mit
+---
+
+# STEEGFormer (small)
+
+ViT-MAE EEG foundation model — braindecode port of **ST-EEGFormer** (small variant).
+
+## Provenance
+
+- **Weights ported from:** [LiuyinYang1101/STEEGFormer](https://github.com/LiuyinYang1101/STEEGFormer),
+  release [`ST-EEGFormer-small`](https://github.com/LiuyinYang1101/STEEGFormer/releases/tag/ST-EEGFormer-small)
+  (asset `checkpoint-300.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 |
+|---|---|---|---|---|---|
+| small | 512 | 8 | 8 | 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-small",
+    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": 512,
+  "depth": 8,
+  "num_heads": 8,
+  "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:2083e58001cf4e90995445e6e8ad1c7531b6564d9a3b7faefe1bfa62c79bee28
+size 101235376

pytorch_model.bin

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