Signal-JEPA pre-training weights (16s, run adeuwv4s, with channel embeddings)

README.md

@@ -0,0 +1,15 @@
+---
+library_name: braindecode
+license: bsd-3-clause
+tags:
+- SignalJEPA
+- _BaseSignalJEPA
+- braindecode
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: https://braindecode.org
+- Paper: [More Information Needed]
+- Docs: https://braindecode.org/stable/generated/braindecode.models.SignalJEPA.html

config.json

@@ -0,0 +1,546 @@
+{
+  "n_outputs": null,
+  "n_chans": null,
+  "chs_info": [
+    {
+      "ch_name": "Fp1",
+      "loc": [
+        -0.0294367,
+        0.0839171,
+        -0.00699
+      ]
+    },
+    {
+      "ch_name": "Fp2",
+      "loc": [
+        0.0298723,
+        0.0848959,
+        -0.00708
+      ]
+    },
+    {
+      "ch_name": "F7",
+      "loc": [
+        -0.0702629,
+        0.0424743,
+        -0.01142
+      ]
+    },
+    {
+      "ch_name": "F3",
+      "loc": [
+        -0.0502438,
+        0.0531112,
+        0.042192
+      ]
+    },
+    {
+      "ch_name": "Fz",
+      "loc": [
+        0.0003122,
+        0.058512,
+        0.066462
+      ]
+    },
+    {
+      "ch_name": "F4",
+      "loc": [
+        0.0518362,
+        0.0543048,
+        0.040814
+      ]
+    },
+    {
+      "ch_name": "F8",
+      "loc": [
+        0.0730431,
+        0.0444217,
+        -0.012
+      ]
+    },
+    {
+      "ch_name": "FC5",
+      "loc": [
+        -0.0772149,
+        0.0186433,
+        0.02446
+      ]
+    },
+    {
+      "ch_name": "FC1",
+      "loc": [
+        -0.0340619,
+        0.0260111,
+        0.079987
+      ]
+    },
+    {
+      "ch_name": "FC2",
+      "loc": [
+        0.0347841,
+        0.0264379,
+        0.078808
+      ]
+    },
+    {
+      "ch_name": "FC6",
+      "loc": [
+        0.0795341,
+        0.0199357,
+        0.024438
+      ]
+    },
+    {
+      "ch_name": "T7",
+      "loc": [
+        -0.0841611,
+        -0.0160187,
+        -0.009346
+      ]
+    },
+    {
+      "ch_name": "C3",
+      "loc": [
+        -0.0653581,
+        -0.0116317,
+        0.064358
+      ]
+    },
+    {
+      "ch_name": "Cz",
+      "loc": [
+        0.0004009,
+        -0.009167,
+        0.100244
+      ]
+    },
+    {
+      "ch_name": "C4",
+      "loc": [
+        0.0671179,
+        -0.0109003,
+        0.06358
+      ]
+    },
+    {
+      "ch_name": "T8",
+      "loc": [
+        0.0850799,
+        -0.0150203,
+        -0.00949
+      ]
+    },
+    {
+      "ch_name": "TP9",
+      "loc": [
+        -0.0856192,
+        -0.0465147,
+        -0.045707
+      ]
+    },
+    {
+      "ch_name": "CP5",
+      "loc": [
+        -0.0795922,
+        -0.0465507,
+        0.030949
+      ]
+    },
+    {
+      "ch_name": "CP1",
+      "loc": [
+        -0.0355131,
+        -0.0472919,
+        0.091315
+      ]
+    },
+    {
+      "ch_name": "CP2",
+      "loc": [
+        0.0383838,
+        -0.0470731,
+        0.090695
+      ]
+    },
+    {
+      "ch_name": "CP6",
+      "loc": [
+        0.0833218,
+        -0.0461013,
+        0.031206
+      ]
+    },
+    {
+      "ch_name": "TP10",
+      "loc": [
+        0.0861618,
+        -0.0470353,
+        -0.045869
+      ]
+    },
+    {
+      "ch_name": "P7",
+      "loc": [
+        -0.0724343,
+        -0.0734527,
+        -0.002487
+      ]
+    },
+    {
+      "ch_name": "P3",
+      "loc": [
+        -0.0530073,
+        -0.0787878,
+        0.05594
+      ]
+    },
+    {
+      "ch_name": "Pz",
+      "loc": [
+        0.0003247,
+        -0.081115,
+        0.082615
+      ]
+    },
+    {
+      "ch_name": "P4",
+      "loc": [
+        0.0556667,
+        -0.0785602,
+        0.056561
+      ]
+    },
+    {
+      "ch_name": "P8",
+      "loc": [
+        0.0730557,
+        -0.0730683,
+        -0.00254
+      ]
+    },
+    {
+      "ch_name": "PO9",
+      "loc": [
+        -0.0549104,
+        -0.0980448,
+        -0.035465
+      ]
+    },
+    {
+      "ch_name": "O1",
+      "loc": [
+        -0.0294134,
+        -0.112449,
+        0.008839
+      ]
+    },
+    {
+      "ch_name": "Oz",
+      "loc": [
+        0.0001076,
+        -0.114892,
+        0.014657
+      ]
+    },
+    {
+      "ch_name": "O2",
+      "loc": [
+        0.0298426,
+        -0.112156,
+        0.0088
+      ]
+    },
+    {
+      "ch_name": "PO10",
+      "loc": [
+        0.0549876,
+        -0.0980911,
+        -0.035541
+      ]
+    },
+    {
+      "ch_name": "FC3",
+      "loc": [
+        -0.0601819,
+        0.0227162,
+        0.055544
+      ]
+    },
+    {
+      "ch_name": "FC4",
+      "loc": [
+        0.0622931,
+        0.0237228,
+        0.05563
+      ]
+    },
+    {
+      "ch_name": "C5",
+      "loc": [
+        -0.0802801,
+        -0.0137597,
+        0.02916
+      ]
+    },
+    {
+      "ch_name": "C1",
+      "loc": [
+        -0.036158,
+        -0.0099839,
+        0.089752
+      ]
+    },
+    {
+      "ch_name": "C2",
+      "loc": [
+        0.037672,
+        -0.0096241,
+        0.088412
+      ]
+    },
+    {
+      "ch_name": "C6",
+      "loc": [
+        0.0834559,
+        -0.0127763,
+        0.029208
+      ]
+    },
+    {
+      "ch_name": "CP3",
+      "loc": [
+        -0.0635562,
+        -0.0470088,
+        0.065624
+      ]
+    },
+    {
+      "ch_name": "CPz",
+      "loc": [
+        0.0003858,
+        -0.047318,
+        0.099432
+      ]
+    },
+    {
+      "ch_name": "CP4",
+      "loc": [
+        0.0666118,
+        -0.0466372,
+        0.06558
+      ]
+    },
+    {
+      "ch_name": "P1",
+      "loc": [
+        -0.0286203,
+        -0.0805249,
+        0.075436
+      ]
+    },
+    {
+      "ch_name": "P2",
+      "loc": [
+        0.0319197,
+        -0.0804871,
+        0.076716
+      ]
+    },
+    {
+      "ch_name": "POz",
+      "loc": [
+        0.0002156,
+        -0.102178,
+        0.050608
+      ]
+    },
+    {
+      "ch_name": "FT9",
+      "loc": [
+        -0.0840759,
+        0.0145673,
+        -0.050429
+      ]
+    },
+    {
+      "ch_name": "FTT9h",
+      "loc": [
+        -0.084125,
+        -0.0018467,
+        -0.029794
+      ]
+    },
+    {
+      "ch_name": "TTP7h",
+      "loc": [
+        -0.0855651,
+        -0.0306287,
+        0.011153
+      ]
+    },
+    {
+      "ch_name": "TP7",
+      "loc": [
+        -0.0848302,
+        -0.0460217,
+        -0.007056
+      ]
+    },
+    {
+      "ch_name": "TPP9h",
+      "loc": [
+        -0.0781602,
+        -0.0607567,
+        -0.023824
+      ]
+    },
+    {
+      "ch_name": "FT10",
+      "loc": [
+        0.0841131,
+        0.0143647,
+        -0.050538
+      ]
+    },
+    {
+      "ch_name": "FTT10h",
+      "loc": [
+        0.084123,
+        -0.0018083,
+        -0.029638
+      ]
+    },
+    {
+      "ch_name": "TPP8h",
+      "loc": [
+        0.0785198,
+        -0.0604323,
+        0.012902
+      ]
+    },
+    {
+      "ch_name": "TP8",
+      "loc": [
+        0.0855488,
+        -0.0455453,
+        -0.00713
+      ]
+    },
+    {
+      "ch_name": "TPP10h",
+      "loc": [
+        0.0789027,
+        -0.0609553,
+        -0.023805
+      ]
+    },
+    {
+      "ch_name": "F9",
+      "loc": [
+        -0.0701019,
+        0.0416523,
+        -0.049952
+      ]
+    },
+    {
+      "ch_name": "F10",
+      "loc": [
+        0.0721141,
+        0.0420667,
+        -0.050452
+      ]
+    },
+    {
+      "ch_name": "AF7",
+      "loc": [
+        -0.0548397,
+        0.0685722,
+        -0.01059
+      ]
+    },
+    {
+      "ch_name": "AF3",
+      "loc": [
+        -0.0337007,
+        0.0768371,
+        0.021227
+      ]
+    },
+    {
+      "ch_name": "AF4",
+      "loc": [
+        0.0357123,
+        0.0777259,
+        0.021956
+      ]
+    },
+    {
+      "ch_name": "AF8",
+      "loc": [
+        0.0557433,
+        0.0696568,
+        -0.010755
+      ]
+    },
+    {
+      "ch_name": "PO3",
+      "loc": [
+        -0.0365114,
+        -0.1008529,
+        0.037167
+      ]
+    },
+    {
+      "ch_name": "PO4",
+      "loc": [
+        0.0367816,
+        -0.1008491,
+        0.036397
+      ]
+    }
+  ],
+  "n_times": null,
+  "input_window_seconds": null,
+  "sfreq": 128.0,
+  "feature_encoder__conv_layers_spec": [
+    [
+      8,
+      32,
+      8
+    ],
+    [
+      16,
+      2,
+      2
+    ],
+    [
+      32,
+      2,
+      2
+    ],
+    [
+      64,
+      2,
+      2
+    ],
+    [
+      64,
+      2,
+      2
+    ]
+  ],
+  "drop_prob": 0.0,
+  "feature_encoder__mode": "default",
+  "feature_encoder__conv_bias": false,
+  "activation": "torch.nn.modules.activation.GELU",
+  "pos_encoder__spat_dim": 30,
+  "pos_encoder__time_dim": 34,
+  "pos_encoder__sfreq_features": 1.0,
+  "pos_encoder__spat_kwargs": null,
+  "transformer__d_model": 64,
+  "transformer__num_encoder_layers": 8,
+  "transformer__num_decoder_layers": 4,
+  "transformer__nhead": 8,
+  "channel_embedding": "pretrain_aligned",
+  "braindecode_version": "1.5.0dev0"
+}

model.safetensors

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

pytorch_model.bin

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