Braindecode 1.5.0.dev984__py3-none-any.whl → 1.5.0.dev985__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (16) hide show
  1. braindecode/models/__init__.py +2 -0
  2. braindecode/models/biot.py +1 -1
  3. braindecode/models/codebrain.py +977 -0
  4. braindecode/models/config.py +5 -4
  5. braindecode/models/signal_jepa.py +446 -38
  6. braindecode/models/summary.csv +1 -0
  7. braindecode/models/util.py +5 -0
  8. braindecode/modules/attention.py +45 -16
  9. braindecode/preprocessing/eegprep_preprocess.py +51 -6
  10. braindecode/version.py +1 -1
  11. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/METADATA +1 -1
  12. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/RECORD +16 -15
  13. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/WHEEL +0 -0
  14. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/licenses/LICENSE.txt +0 -0
  15. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/licenses/NOTICE.txt +0 -0
  16. {braindecode-1.5.0.dev984.dist-info → braindecode-1.5.0.dev985.dist-info}/top_level.txt +0 -0
braindecode/models/__init__.py CHANGED
@@ -8,6 +8,7 @@ from .bendr import BENDR
8
8
  from .biot import BIOT
9
9
  from .brainmodule import BrainModule
10
10
  from .cbramod import CBraMod
11
+ from .codebrain import CodeBrain
11
12
  from .contrawr import ContraWR
12
13
  from .ctnet import CTNet
13
14
  from .deep4 import Deep4Net
@@ -72,6 +73,7 @@ __all__ = [
72
73
  "BIOT",
73
74
  "BENDR",
74
75
  "CBraMod",
76
+ "CodeBrain",
75
77
  "ContraWR",
76
78
  "CTNet",
77
79
  "Deep4Net",
braindecode/models/biot.py CHANGED
@@ -439,7 +439,7 @@ class _BIOTEncoder(nn.Module):
439
439
  self.channel_tokens = nn.Embedding(
440
440
  num_embeddings=n_chans, embedding_dim=emb_size
441
441
  )
442
- self.index = nn.Parameter(torch.LongTensor(range(n_chans)), requires_grad=False)
442
+ self.register_buffer("index", torch.arange(n_chans, dtype=torch.long))
443
443
 
444
444
  def stft(self, sample):
445
445
  """