braindecode 1.5.0.dev984__tar.gz → 1.5.0.dev174273210__tar.gz

{braindecode-1.5.0.dev984/braindecode.egg-info → braindecode-1.5.0.dev174273210}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.0.dev984
+Version: 1.5.0.dev174273210
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Alexandre Gramfort <agramfort@meta.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>

{braindecode-1.5.0.dev984 → braindecode-1.5.0.dev174273210}/braindecode/models/__init__.py

@@ -8,6 +8,7 @@ from .bendr import BENDR
 from .biot import BIOT
 from .brainmodule import BrainModule
 from .cbramod import CBraMod
+from .codebrain import CodeBrain
 from .contrawr import ContraWR
 from .ctnet import CTNet
 from .deep4 import Deep4Net
@@ -72,6 +73,7 @@ __all__ = [
     "BIOT",
     "BENDR",
     "CBraMod",
+    "CodeBrain",
     "ContraWR",
     "CTNet",
     "Deep4Net",

{braindecode-1.5.0.dev984 → braindecode-1.5.0.dev174273210}/braindecode/models/biot.py

@@ -439,7 +439,7 @@ class _BIOTEncoder(nn.Module):
         self.channel_tokens = nn.Embedding(
             num_embeddings=n_chans, embedding_dim=emb_size
         )
-        self.index = nn.Parameter(torch.LongTensor(range(n_chans)), requires_grad=False)
+        self.register_buffer("index", torch.arange(n_chans, dtype=torch.long))
 
     def stft(self, sample):
         """