braindecode 0.4.5__tar.gz → 0.4.6__tar.gz

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 (77) hide show
  1. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/PKG-INFO +1 -1
  2. {Braindecode-0.4.5 → Braindecode-0.4.6}/PKG-INFO +1 -1
  3. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/base.py +6 -2
  4. Braindecode-0.4.6/braindecode/version.py +1 -0
  5. Braindecode-0.4.5/braindecode/version.py +0 -1
  6. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/SOURCES.txt +0 -0
  7. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/dependency_links.txt +0 -0
  8. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/not-zip-safe +0 -0
  9. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/requires.txt +0 -0
  10. {Braindecode-0.4.5 → Braindecode-0.4.6}/Braindecode.egg-info/top_level.txt +0 -0
  11. {Braindecode-0.4.5 → Braindecode-0.4.6}/LICENSE.txt +0 -0
  12. {Braindecode-0.4.5 → Braindecode-0.4.6}/MANIFEST.in +0 -0
  13. {Braindecode-0.4.5 → Braindecode-0.4.6}/README.rst +0 -0
  14. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/__init__.py +0 -0
  15. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datasets/__init__.py +0 -0
  16. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datasets/bbci.py +0 -0
  17. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datasets/bcic_iv_2a.py +0 -0
  18. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datasets/multiple.py +0 -0
  19. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datasets/sensor_positions.py +0 -0
  20. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/__init__.py +0 -0
  21. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/iterators.py +0 -0
  22. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/signal_target.py +0 -0
  23. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/signalproc.py +0 -0
  24. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/splitters.py +0 -0
  25. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/trial_segment.py +0 -0
  26. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/datautil/util.py +0 -0
  27. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/experiments/__init__.py +0 -0
  28. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/experiments/experiment.py +0 -0
  29. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/experiments/loggers.py +0 -0
  30. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/experiments/monitors.py +0 -0
  31. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/experiments/stopcriteria.py +0 -0
  32. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/mne_ext/__init__.py +0 -0
  33. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/mne_ext/signalproc.py +0 -0
  34. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/__init__.py +0 -0
  35. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/deep4.py +0 -0
  36. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/eegnet.py +0 -0
  37. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/hybrid.py +0 -0
  38. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/shallow_fbcsp.py +0 -0
  39. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/models/util.py +0 -0
  40. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/__init__.py +0 -0
  41. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/constraints.py +0 -0
  42. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/functions.py +0 -0
  43. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/init.py +0 -0
  44. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/losses.py +0 -0
  45. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/modules.py +0 -0
  46. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/optimizers.py +0 -0
  47. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/schedulers.py +0 -0
  48. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/torch_ext/util.py +0 -0
  49. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/util.py +0 -0
  50. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/visualization/__init__.py +0 -0
  51. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/visualization/input_windows.py +0 -0
  52. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/visualization/perturbation.py +0 -0
  53. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/visualization/plot.py +0 -0
  54. {Braindecode-0.4.5 → Braindecode-0.4.6}/braindecode/visualization/sinfit.py +0 -0
  55. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/Makefile +0 -0
  56. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/conf.py +0 -0
  57. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/index.rst +0 -0
  58. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/BBCI_Data.ipynb +0 -0
  59. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/BBCI_Data_Start_Stop.ipynb +0 -0
  60. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/BBCI_Data_Trialwise.ipynb +0 -0
  61. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/Cropped_Decoding.ipynb +0 -0
  62. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/Cropped_Manual_Training_Loop.ipynb +0 -0
  63. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/Cropped_Trialwise_Explanation_Figures.ipynb +0 -0
  64. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/Trialwise_Decoding.ipynb +0 -0
  65. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/Trialwise_Manual_Training_Loop.ipynb +0 -0
  66. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/notebooks/visualization/Perturbation.ipynb +0 -0
  67. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.datasets.rst +0 -0
  68. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.datautil.rst +0 -0
  69. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.experiments.rst +0 -0
  70. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.mne_ext.rst +0 -0
  71. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.models.rst +0 -0
  72. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.rst +0 -0
  73. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.torch_ext.rst +0 -0
  74. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/braindecode.visualization.rst +0 -0
  75. {Braindecode-0.4.5 → Braindecode-0.4.6}/docs/source/modules.rst +0 -0
  76. {Braindecode-0.4.5 → Braindecode-0.4.6}/setup.cfg +0 -0
  77. {Braindecode-0.4.5 → Braindecode-0.4.6}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 1.1
2
2
  Name: Braindecode
3
- Version: 0.4.5
3
+ Version: 0.4.6
4
4
  Summary: A deep learning toolbox to decode raw time-domain EEG.
5
5
  Home-page: https://github.com/robintibor/braindecode
6
6
  Author: Robin Tibor Schirrmeister
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 1.1
2
2
  Name: Braindecode
3
- Version: 0.4.5
3
+ Version: 0.4.6
4
4
  Summary: A deep learning toolbox to decode raw time-domain EEG.
5
5
  Home-page: https://github.com/robintibor/braindecode
6
6
  Author: Robin Tibor Schirrmeister
@@ -285,8 +285,12 @@ class BaseModel(object):
285
285
  Predicted labels per trial.
286
286
  """
287
287
  all_preds = []
288
- for b_X, _ in self.iterator.get_batches(SignalAndTarget(X, X), False):
289
- all_preds.append(var_to_np(self.network(np_to_var(b_X))))
288
+ with th.no_grad():
289
+ for b_X, _ in self.iterator.get_batches(SignalAndTarget(X, X), False):
290
+ b_X_var = np_to_var(b_X)
291
+ if self.cuda:
292
+ b_X_var = b_X_var.cuda()
293
+ all_preds.append(var_to_np(self.network(b_X_var)))
290
294
  if self.cropped:
291
295
  pred_labels = compute_trial_labels_from_crop_preds(
292
296
  all_preds, self.iterator.input_time_length, X)
@@ -0,0 +1 @@
1
+ __version__ = "0.4.6"
@@ -1 +0,0 @@
1
- __version__ = "0.4.5"
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes