Add CodeBrain pretrained weights (TUEG, 19ch 10-20, 200Hz)

README.md

@@ -0,0 +1,14 @@
+---
+library_name: braindecode
+license: bsd-3-clause
+tags:
+- CodeBrain
+- 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.CodeBrain.html

config.json

@@ -0,0 +1,112 @@
+{
+  "n_outputs": 2,
+  "n_chans": 19,
+  "chs_info": [
+    {
+      "ch_name": "C3",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "C4",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "Cz",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "F3",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "F4",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "Fp1",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "Fp2",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "F7",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "F8",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "Fz",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "O1",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "O2",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "P3",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "P4",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "Pz",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "T3",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "T4",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "T5",
+      "kind": "eeg"
+    },
+    {
+      "ch_name": "T6",
+      "kind": "eeg"
+    }
+  ],
+  "n_times": 6000,
+  "input_window_seconds": 30.0,
+  "sfreq": 200.0,
+  "patch_size": 200,
+  "res_channels": 200,
+  "skip_channels": 200,
+  "out_channels": 200,
+  "num_res_layers": 8,
+  "drop_prob": 0.1,
+  "s4_bidirectional": true,
+  "s4_layernorm": false,
+  "s4_lmax": 570,
+  "s4_d_state": 64,
+  "conv_out_chans": 25,
+  "conv_groups": 5,
+  "proj_kernel_size": 49,
+  "proj_padding": 24,
+  "proj_refine_kernel": 3,
+  "pos_kernel": [
+    19,
+    7
+  ],
+  "spectral_dropout": 0.1,
+  "mlp_hidden_multiplier": 4,
+  "swa_window_size": 1,
+  "codebook_size_t": 4096,
+  "codebook_size_f": 4096,
+  "pretrain_mode": false,
+  "activation": "torch.nn.modules.activation.ReLU",
+  "braindecode_version": "1.3.2"
+}

model.safetensors

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

pytorch_model.bin

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