Signal-JEPA pre-training weights (16s, run adeuwv4s, without 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,47 @@
+{
+  "n_outputs": null,
+  "n_chans": null,
+  "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,
+  "braindecode_version": "1.5.0dev0"
+}

model.safetensors

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

pytorch_model.bin

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