braindecode 1.2.0.dev168908090__py3-none-any.whl → 1.2.0.dev175267687__py3-none-any.whl

{braindecode-1.2.0.dev168908090.dist-info → braindecode-1.2.0.dev175267687.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.2.0.dev168908090
+Version: 1.2.0.dev175267687
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
@@ -17,7 +17,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >3.10
+Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 License-File: NOTICE.txt
@@ -49,6 +49,10 @@ Requires-Dist: mypy; extra == "tests"
 Provides-Extra: docs
 Requires-Dist: sphinx_gallery; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
+Requires-Dist: sphinx-autobuild; extra == "docs"
+Requires-Dist: sphinxcontrib-bibtex; extra == "docs"
+Requires-Dist: sphinx_sitemap; extra == "docs"
 Requires-Dist: pydata_sphinx_theme; extra == "docs"
 Requires-Dist: numpydoc; extra == "docs"
 Requires-Dist: memory_profiler; extra == "docs"
@@ -59,6 +63,7 @@ Requires-Dist: lightning; extra == "docs"
 Requires-Dist: seaborn; extra == "docs"
 Requires-Dist: pre-commit; extra == "docs"
 Requires-Dist: openneuro-py; extra == "docs"
+Requires-Dist: plotly; extra == "docs"
 Provides-Extra: all
 Requires-Dist: braindecode[docs,moabb,tests]; extra == "all"
 Dynamic: license-file

{braindecode-1.2.0.dev168908090.dist-info → braindecode-1.2.0.dev175267687.dist-info}/RECORD

@@ -3,7 +3,7 @@ braindecode/classifier.py,sha256=k9vSCtfQbld0YVleDi5rrrmk6k_k5JYEPPBYcNxYjZ8,980
 braindecode/eegneuralnet.py,sha256=dz8k_-2jV7WqkaX4bQG-dmr-vRT7ZtOwJqomXyC9PTw,15287
 braindecode/regressor.py,sha256=VLfrpiXklwI4onkwue3QmzlBWcvspu0tlrLo9RT1Oiw,9375
 braindecode/util.py,sha256=J-tBcDJNlMTIFW2mfOy6Ko0nsgdP4obRoEVDeg2rFH0,12686
-braindecode/version.py,sha256=b8BdgKb6YgI65s6lSGpGOwj1zjSE3ObTFYtA1t6_lrU,35
+braindecode/version.py,sha256=DjAGynn_F-_dyPfKBBkITwv_h_JYqDSa0IISxiwpVAE,35
 braindecode/augmentation/__init__.py,sha256=LG7ONqCufYAF9NZt8POIp10lYXb8iSueYkF-CWGK2Ls,1001
 braindecode/augmentation/base.py,sha256=gg7wYsVfa9jfqBddtE03B5ZrPHFFmPl2sa3LOrRnGfo,7325
 braindecode/augmentation/functional.py,sha256=ygkMNEFHaUdRQfk7meMML19FnM406Uf34h-ztKXdJwM,37978
@@ -13,6 +13,7 @@ braindecode/datasets/base.py,sha256=ED8RQWusMyWf0T7b_HXwouR2Ax47qppEc506AlSzBt0,
 braindecode/datasets/bbci.py,sha256=BC9o1thEyYBREAo930O7zZz3xZB-l4Odt5j8E_1huXI,19277
 braindecode/datasets/bcicomp.py,sha256=ER_XmqxhpoO2FWELMesQXQ40OTe7BXoy7nYDSiZG9kE,7556
 braindecode/datasets/bids.py,sha256=4asq1HyQHgJjwW7w-GMlvTVQhi-hR2HWLJ8Z__UrUS4,8846
+braindecode/datasets/experimental.py,sha256=Z_uzMNA875-l878LAv7bWiWYJX3QAefmb5quBkcPp7M,8514
 braindecode/datasets/mne.py,sha256=Dg6RZAAwd8TVGrvLOPF5B_JrbyGUWg52vWmn6fLMOQM,6135
 braindecode/datasets/moabb.py,sha256=JmBcFV7QJT8GCgLNNKWgxJVnEVnO5wd9U_uiIqTIxDM,7091
 braindecode/datasets/nmt.py,sha256=E4T8OYBEwWRSjh7VFzmyxaZbf5ufFVEBYYmQEd1ghUU,10430
@@ -26,25 +27,25 @@ braindecode/datautil/util.py,sha256=ZfDoxLieKsgI8xcWQqebV-vJ5pJYRvRRHkEwhwpgoKU,
 braindecode/functional/__init__.py,sha256=JPUDFeKtfogEzfrwPaZRBmxexPjBw7AglYMlImaAnWc,413
 braindecode/functional/functions.py,sha256=CoEweM6YLhigx0tNmmz6yAc8iQ078sTFY2GeCjK5fFs,8622
 braindecode/functional/initialization.py,sha256=BUSC7y2TMsfShpMYBVwm3xg3ODFqWp-STH7yD4sn8zk,1388
-braindecode/models/__init__.py,sha256=xv1QPELZxocPgbc_mz-eYM5w08ZDNOsDV4pOnIFhUww,2551
-braindecode/models/atcnet.py,sha256=Pn5KzQjv7YxSNDr_CY6O_Yg9K4m9XJ7btCIqyzkcPxc,32102
-braindecode/models/attentionbasenet.py,sha256=1uwrtsdEGiBwokkO8A_2SR5zapOTQUBZd4q7hIpR0cw,23359
+braindecode/models/__init__.py,sha256=v2Pn0H-rM_9xr1EEoKIFygmhbS9r52qh8XwFzXuhK70,2455
+braindecode/models/atcnet.py,sha256=jA_18BOaasmiqGbLJOvfBY5q2xHtKdoRFKzN_aqpDoQ,32107
+braindecode/models/attentionbasenet.py,sha256=AK78VvwrZXyJY20zadzDUHl17C-5zcWCd5xPRN7Lr4o,30385
+braindecode/models/attn_sleep.py,sha256=m6sdFfD4en2hHf_TpotLPC1hVweJcYZvjgf12bV5FZg,17822
 braindecode/models/base.py,sha256=9icrWNZBGbh_VLyB9m8g_K1QyK7s3mh8X-hJ29gEbWs,10802
 braindecode/models/biot.py,sha256=T4PymX3penMJcrdfb5Nq6B3P-jyP2laAIu_R9o3uCXo,17512
 braindecode/models/contrawr.py,sha256=eeR_ik4gNZ3rJLM6Mw9gJ2gTMkZ8CU8C4rN_GQMQTAE,10044
-braindecode/models/ctnet.py,sha256=-J9QtUM8kcntz_xinfuBBvwDMECHiMPMcr2MS4GDPEY,17308
-braindecode/models/deep4.py,sha256=YJQUw-0EuFUi4qjm8caJGB8wRM_aeJa5X_d8jrGaQAI,14588
-braindecode/models/deepsleepnet.py,sha256=RrciuVJtZ-fhiUl-yLPfK2FP-G29V5Wor6pPlrMHQWQ,9218
-braindecode/models/eegconformer.py,sha256=6scz0Axm97JV7-4u5yd6HGE7PldAMR39x5qSNzjSqxQ,17404
-braindecode/models/eeginception_erp.py,sha256=mwh3rGSHAJVvnbOlYTuWWkKxlmFAdAXBNCrq4IPgOS4,11408
-braindecode/models/eeginception_mi.py,sha256=aKJRFuYrpbcRbmmT2xVghKbK8pnl7fzu5hrV0ybRKso,12424
+braindecode/models/ctnet.py,sha256=ce5F31q2weBKvg7PL80iDm7za9fhGaCFvNfHoJW_dtg,17315
+braindecode/models/deep4.py,sha256=-s-R3H7so2xlSiPsU226eSwscv1X9xJMYLm3LhZ3mSU,14645
+braindecode/models/deepsleepnet.py,sha256=wGSAXW73Ga1-HFbn7kXiLeGsJceiqZyMLZnX2UZZXWw,15207
+braindecode/models/eegconformer.py,sha256=rxMAmqErDVLq7nS77CnTtpcC3C2OR_EoZ8-jG-dKP9I,17433
+braindecode/models/eeginception_erp.py,sha256=FYXoM-u4kOodMzGgvKDn7IwJwHl9Z0iiWx9bVHiO9EY,16324
+braindecode/models/eeginception_mi.py,sha256=VoWtsaWj1xQ4FlrvCbnPvo8eosufYUmTrL4uvFtqKcg,12456
 braindecode/models/eegitnet.py,sha256=feXFmPCd-Ejxt7jgWPen1Ag0-oSclDVQai0Atwu9d_A,9827
 braindecode/models/eegminer.py,sha256=ouKZah9Q7_sxT7DJJMcPObwVxNQE87sEljJg6QwiQNw,9847
-braindecode/models/eegnet.py,sha256=YeBCmU6Al9FDS4MZQTOLd0MCUfPbM6tmVlGWpb59Qzg,19256
-braindecode/models/eegnex.py,sha256=KNJIh8pFNhY087Bey2OPzDD4Uqw9pS6UkwMjnOngBzg,8497
-braindecode/models/eegresnet.py,sha256=cqWOSGqfJN_dNYUU9l8nYd_S3T1N-UX5-encKQzfBlg,12057
-braindecode/models/eegsimpleconv.py,sha256=sHpK-7ZGOCMuXsdkSVuarFTd1T0jMJUP_xwXP3gxQwc,7268
-braindecode/models/eegtcnet.py,sha256=np-93Ttctp2uaEYpMrfXfH5bJmCOUZZHLjv8GJEEym4,10830
+braindecode/models/eegnet.py,sha256=dIaHZoz7xMII1qKrS0___IWdy1xg2QrMMiqUgTJM9E8,13682
+braindecode/models/eegnex.py,sha256=eahHolFl15LwNWeC5qjQqUGqURibQZIV425rI1p-dG8,13604
+braindecode/models/eegsimpleconv.py,sha256=6V5ZQNWijmd3-2wv7lJB_HGBS3wHWWVrKoNIeWTXu-w,7300
+braindecode/models/eegtcnet.py,sha256=Y53uJEX_hoB6eHCew9SIfzNxCYea8UhljDARJTk-Tq8,10837
 braindecode/models/fbcnet.py,sha256=RBCLOaiUvivfsT2mq6FN0Kp1-rR3iB0ElzVpHxRl4oI,7486
 braindecode/models/fblightconvnet.py,sha256=d5MwhawhkjilAMo0ckaYMxJhdGMEuorWgHX-TBgwv6s,11041
 braindecode/models/fbmsnet.py,sha256=9bZn2_n1dTrI1Qh3Sz9zMZnH_a-Yq-13UHYSmF6r_UE,11659
@@ -52,21 +53,20 @@ braindecode/models/hybrid.py,sha256=hA8jwD3_3LL71BxUjRM1dkhqlHU9E9hjuDokh-jBq-4,
 braindecode/models/ifnet.py,sha256=Y2bwfko3SDjD74AzgUEzgMhKJFGCCw_Q_Noh5VONEjQ,15137
 braindecode/models/labram.py,sha256=vcrpwiu4F-djtIPscFbtP2Y0jTosyR_cXnOMQQRGPLw,41798
 braindecode/models/msvtnet.py,sha256=hxeCLkHS6w2w89YlLfEPCyQ4XQQpt45bEYPiQJ9SFzY,12642
-braindecode/models/sccnet.py,sha256=baGsNpVRdyWzbkTizOthJoJGejLb8BxMpN9ODwZinio,7919
-braindecode/models/shallow_fbcsp.py,sha256=-sL6XCmCUZVhKKrC84-KWgwhWKQQvev1oNSmH_d6FA4,7499
+braindecode/models/sccnet.py,sha256=ragbEzNrua0S84H4JR_j2QGLZWrFKGQ4CfIS2epIYEk,11919
+braindecode/models/shallow_fbcsp.py,sha256=7U07DJBrm2JHV8v5ja-xuE5-IH5tfmryhJtrfO1n4jk,7531
 braindecode/models/signal_jepa.py,sha256=UeSkeAM3Qmx8bbAqHCj5nP-PtZM00_5SGA8ibo9mptc,37079
 braindecode/models/sinc_shallow.py,sha256=Ilv8K1XhMGiRTBtQdq7L595i6cEFYOBe0_UDv-LqL7s,11907
-braindecode/models/sleep_stager_blanco_2020.py,sha256=qPKMDLuv4J7et4dZHyTe-j0oB6ESYn9mA_aW7RMC-rU,6002
-braindecode/models/sleep_stager_chambon_2018.py,sha256=62x2Rdjd5UZDX8YlnfAtdRCrjLsPvPpnUweGElZLdkw,5213
-braindecode/models/sleep_stager_eldele_2021.py,sha256=-4ISuznykDy9ZFzUM-OeiGCwmgM3U-LuyoDSrhPbRDw,17555
-braindecode/models/sparcnet.py,sha256=eZMoJOxlcIyHPdQiX7KXUKuUBlAWkTwsXNWmNma_KAI,13941
-braindecode/models/summary.csv,sha256=l7HYYwv3Z69JRPVIhVq-wr_nC1J1KIz6IGw_zeRSk58,6110
+braindecode/models/sleep_stager_blanco_2020.py,sha256=vXulnDYutEFLM0UPXyAI0YIj5QImUMVEmYZb78j34H8,6034
+braindecode/models/sleep_stager_chambon_2018.py,sha256=8w8IR2PsfG0jSc3o0YVopgHpOvCHNIuMi7-QRJOYEW4,5245
+braindecode/models/sparcnet.py,sha256=MG1OB91guI7ssKRk8GvWlzUvaxo_otaYnbEGzNUZVyg,13973
+braindecode/models/summary.csv,sha256=NfrmnjyfDmWVe2zyNqgczEQcLI910BOS4sICtcKS3gc,6765
 braindecode/models/syncnet.py,sha256=nrWJC5ijCSWKVZyRn-dmOuc1t5vk2C6tx8U3U4j5d5Y,8362
 braindecode/models/tcn.py,sha256=SQu56H9zdbcbbDIXZVgZtJg7es8CRAJ7z-IBnmf4UWM,8158
-braindecode/models/tidnet.py,sha256=k7Q0yAnEBmq1sqhsvoV4-g8wfYSUQ-C3iYxfLp5m8xQ,11805
-braindecode/models/tsinception.py,sha256=EcfLDDJXZloh_vrKRuxAHYRZ1EVWlEKHNXqybTRrTbQ,10116
-braindecode/models/usleep.py,sha256=dFh3KiZITu13gMxcbPGoK4hq2ySDWzVSCQXkj1006w0,11605
-braindecode/models/util.py,sha256=VrhwG1YBGwKohCej6TmhrNAIoleQHRu3YdiBPuHFY_E,5302
+braindecode/models/tidnet.py,sha256=HSUL1al6gaRbJ-BRYAAs4KDvLuKEvh0NnBfAsPeWMpM,11837
+braindecode/models/tsinception.py,sha256=nnQxzpqRy9FPuN5xgh9fNQ386VbreQ_nZBSFNkSfal0,10356
+braindecode/models/usleep.py,sha256=5uztUHX70T_LurqRob_XmVnKkZDwt74x2Iz181M7s54,17233
+braindecode/models/util.py,sha256=VZGVPhUSsoP47pta0_UhC2-g5n5-EFZAW93ZVccrEHU,5232
 braindecode/modules/__init__.py,sha256=PD2LpeSHWW_MgEef7-G8ief5gheGObzsIoacchxWuyA,1756
 braindecode/modules/activation.py,sha256=lTO2IjZWBDeXZ4ZVDgLmTDmxHdqyAny3Fsy07HY9tmQ,1466
 braindecode/modules/attention.py,sha256=ISE11jXAvMqKpawZilg8i7lDX5mkuvpEplrh_CtGEkk,24102
@@ -93,9 +93,9 @@ braindecode/training/scoring.py,sha256=WRkwqbitA3m_dzRnGp2ZIZPge5Nhx9gAEQhIHzeH4
 braindecode/visualization/__init__.py,sha256=4EER_xHqZIDzEvmgUEm7K1bgNKpyZAIClR9ZCkMuY4M,240
 braindecode/visualization/confusion_matrices.py,sha256=qIWMLEHow5CJ7PhGggD8mnD55Le6xhma9HSzt4R33fc,9509
 braindecode/visualization/gradients.py,sha256=KZo-GA0uwiwty2_94j2IjmCR2SKcfPb1Bi3sQq7vpTk,2170
-braindecode-1.2.0.dev168908090.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
-braindecode-1.2.0.dev168908090.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
-braindecode-1.2.0.dev168908090.dist-info/METADATA,sha256=k8t6kilsNd1r7kcjt85QWco9RF9s4XVwFDXSv3wqv8M,6883
-braindecode-1.2.0.dev168908090.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-braindecode-1.2.0.dev168908090.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
-braindecode-1.2.0.dev168908090.dist-info/RECORD,,
+braindecode-1.2.0.dev175267687.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
+braindecode-1.2.0.dev175267687.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
+braindecode-1.2.0.dev175267687.dist-info/METADATA,sha256=oQss190zuseCAh-RsSLYE_Ak5oLrSIQzh0G2PrdZwNA,7129
+braindecode-1.2.0.dev175267687.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+braindecode-1.2.0.dev175267687.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
+braindecode-1.2.0.dev175267687.dist-info/RECORD,,

braindecode/models/eegresnet.py

@@ -1,362 +0,0 @@
-# Authors: Robin Tibor Schirrmeister <robintibor@gmail.com>
-#          Tonio Ball
-#
-# License: BSD-3
-
-import numpy as np
-import torch
-from einops.layers.torch import Rearrange
-from torch import nn
-from torch.nn import init
-
-from braindecode.models.base import EEGModuleMixin
-from braindecode.modules import (
-    AvgPool2dWithConv,
-    Ensure4d,
-    SqueezeFinalOutput,
-)
-
-
-class EEGResNet(EEGModuleMixin, nn.Sequential):
-    """EEGResNet from Schirrmeister et al. 2017 [Schirrmeister2017]_.
-
-    .. figure:: https://onlinelibrary.wiley.com/cms/asset/bed1b768-809f-4bc6-b942-b36970d81271/hbm23730-fig-0003-m.jpg
-        :align: center
-        :alt: EEGResNet Architecture
-
-    Model described in [Schirrmeister2017]_.
-
-    Parameters
-    ----------
-    in_chans :
-        Alias for ``n_chans``.
-    n_classes :
-        Alias for ``n_outputs``.
-    input_window_samples :
-       Alias for ``n_times``.
-    activation: nn.Module, default=nn.ELU
-        Activation function class to apply. Should be a PyTorch activation
-        module class like ``nn.ReLU`` or ``nn.ELU``. Default is ``nn.ELU``.
-
-    References
-    ----------
-    .. [Schirrmeister2017] Schirrmeister, R. T., Springenberg, J. T., Fiederer,
-       L. D. J., Glasstetter, M., Eggensperger, K., Tangermann, M., Hutter, F.
-       & Ball, T. (2017). Deep learning with convolutional neural networks for ,
-       EEG decoding and visualization. Human Brain Mapping, Aug. 2017.
-       Online: http://dx.doi.org/10.1002/hbm.23730
-    """
-
-    def __init__(
-        self,
-        n_chans=None,
-        n_outputs=None,
-        n_times=None,
-        final_pool_length="auto",
-        n_first_filters=20,
-        n_layers_per_block=2,
-        first_filter_length=3,
-        activation=nn.ELU,
-        split_first_layer=True,
-        batch_norm_alpha=0.1,
-        batch_norm_epsilon=1e-4,
-        conv_weight_init_fn=lambda w: init.kaiming_normal_(w, a=0),
-        chs_info=None,
-        input_window_seconds=None,
-        sfreq=250,
-    ):
-        super().__init__(
-            n_outputs=n_outputs,
-            n_chans=n_chans,
-            chs_info=chs_info,
-            n_times=n_times,
-            input_window_seconds=input_window_seconds,
-            sfreq=sfreq,
-        )
-        del n_outputs, n_chans, chs_info, n_times, input_window_seconds, sfreq
-
-        if final_pool_length == "auto":
-            assert self.n_times is not None
-        assert first_filter_length % 2 == 1
-        self.final_pool_length = final_pool_length
-        self.n_first_filters = n_first_filters
-        self.n_layers_per_block = n_layers_per_block
-        self.first_filter_length = first_filter_length
-        self.nonlinearity = activation
-        self.split_first_layer = split_first_layer
-        self.batch_norm_alpha = batch_norm_alpha
-        self.batch_norm_epsilon = batch_norm_epsilon
-        self.conv_weight_init_fn = conv_weight_init_fn
-
-        self.mapping = {
-            "conv_classifier.weight": "final_layer.conv_classifier.weight",
-            "conv_classifier.bias": "final_layer.conv_classifier.bias",
-        }
-
-        self.add_module("ensuredims", Ensure4d())
-        if self.split_first_layer:
-            self.add_module("dimshuffle", Rearrange("batch C T 1 -> batch 1 T C"))
-            self.add_module(
-                "conv_time",
-                nn.Conv2d(
-                    1,
-                    self.n_first_filters,
-                    (self.first_filter_length, 1),
-                    stride=1,
-                    padding=(self.first_filter_length // 2, 0),
-                ),
-            )
-            self.add_module(
-                "conv_spat",
-                nn.Conv2d(
-                    self.n_first_filters,
-                    self.n_first_filters,
-                    (1, self.n_chans),
-                    stride=(1, 1),
-                    bias=False,
-                ),
-            )
-        else:
-            self.add_module(
-                "conv_time",
-                nn.Conv2d(
-                    self.n_chans,
-                    self.n_first_filters,
-                    (self.first_filter_length, 1),
-                    stride=(1, 1),
-                    padding=(self.first_filter_length // 2, 0),
-                    bias=False,
-                ),
-            )
-        n_filters_conv = self.n_first_filters
-        self.add_module(
-            "bnorm",
-            nn.BatchNorm2d(
-                n_filters_conv, momentum=self.batch_norm_alpha, affine=True, eps=1e-5
-            ),
-        )
-        self.add_module("conv_nonlin", self.nonlinearity())
-        cur_dilation = np.array([1, 1])
-        n_cur_filters = n_filters_conv
-        i_block = 1
-        for i_layer in range(self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-        i_block += 1
-        cur_dilation[0] *= 2
-        n_out_filters = int(2 * n_cur_filters)
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_out_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        n_cur_filters = n_out_filters
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-
-        i_block += 1
-        cur_dilation[0] *= 2
-        n_out_filters = int(1.5 * n_cur_filters)
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_out_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        n_cur_filters = n_out_filters
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-
-        i_block += 1
-        cur_dilation[0] *= 2
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_cur_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-
-        i_block += 1
-        cur_dilation[0] *= 2
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_cur_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-
-        i_block += 1
-        cur_dilation[0] *= 2
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_cur_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-        i_block += 1
-        cur_dilation[0] *= 2
-        self.add_module(
-            "res_{:d}_{:d}".format(i_block, 0),
-            _ResidualBlock(
-                n_cur_filters,
-                n_cur_filters,
-                dilation=cur_dilation,
-            ),
-        )
-        for i_layer in range(1, self.n_layers_per_block):
-            self.add_module(
-                "res_{:d}_{:d}".format(i_block, i_layer),
-                _ResidualBlock(n_cur_filters, n_cur_filters, dilation=cur_dilation),
-            )
-
-        self.eval()
-        if self.final_pool_length == "auto":
-            self.add_module("mean_pool", nn.AdaptiveAvgPool2d((1, 1)))
-        else:
-            pool_dilation = int(cur_dilation[0]), int(cur_dilation[1])
-            self.add_module(
-                "mean_pool",
-                AvgPool2dWithConv(
-                    (self.final_pool_length, 1), (1, 1), dilation=pool_dilation
-                ),
-            )
-
-        # Incorporating classification module and subsequent ones in one final layer
-        module = nn.Sequential()
-
-        module.add_module(
-            "conv_classifier",
-            nn.Conv2d(
-                n_cur_filters,
-                self.n_outputs,
-                (1, 1),
-                bias=True,
-            ),
-        )
-
-        module.add_module("squeeze", SqueezeFinalOutput())
-
-        self.add_module("final_layer", module)
-
-        # Initialize all weights
-        self.apply(lambda module: self._weights_init(module, self.conv_weight_init_fn))
-
-        # Start in train mode
-        self.train()
-
-    @staticmethod
-    def _weights_init(module, conv_weight_init_fn):
-        """
-        initialize weights
-        """
-        classname = module.__class__.__name__
-        if "Conv" in classname and classname != "AvgPool2dWithConv":
-            conv_weight_init_fn(module.weight)
-            if module.bias is not None:
-                init.constant_(module.bias, 0)
-        elif "BatchNorm" in classname:
-            init.constant_(module.weight, 1)
-            init.constant_(module.bias, 0)
-
-
-class _ResidualBlock(nn.Module):
-    """
-    create a residual learning building block with two stacked 3x3 convlayers as in paper
-    """
-
-    def __init__(
-        self,
-        in_filters,
-        out_num_filters,
-        dilation,
-        filter_time_length=3,
-        nonlinearity: nn.Module = nn.ELU,
-        batch_norm_alpha=0.1,
-        batch_norm_epsilon=1e-4,
-    ):
-        super(_ResidualBlock, self).__init__()
-        time_padding = int((filter_time_length - 1) * dilation[0])
-        assert time_padding % 2 == 0
-        time_padding = int(time_padding // 2)
-        dilation = (int(dilation[0]), int(dilation[1]))
-        assert (out_num_filters - in_filters) % 2 == 0, (
-            "Need even number of extra channels in order to be able to pad correctly"
-        )
-        self.n_pad_chans = out_num_filters - in_filters
-
-        self.conv_1 = nn.Conv2d(
-            in_filters,
-            out_num_filters,
-            (filter_time_length, 1),
-            stride=(1, 1),
-            dilation=dilation,
-            padding=(time_padding, 0),
-        )
-        self.bn1 = nn.BatchNorm2d(
-            out_num_filters,
-            momentum=batch_norm_alpha,
-            affine=True,
-            eps=batch_norm_epsilon,
-        )
-        self.conv_2 = nn.Conv2d(
-            out_num_filters,
-            out_num_filters,
-            (filter_time_length, 1),
-            stride=(1, 1),
-            dilation=dilation,
-            padding=(time_padding, 0),
-        )
-        self.bn2 = nn.BatchNorm2d(
-            out_num_filters,
-            momentum=batch_norm_alpha,
-            affine=True,
-            eps=batch_norm_epsilon,
-        )
-        # also see https://mail.google.com/mail/u/0/#search/ilya+joos/1576137dd34c3127
-        # for resnet options as ilya used them
-        self.nonlinearity = nonlinearity()
-
-    def forward(self, x):
-        stack_1 = self.nonlinearity(self.bn1(self.conv_1(x)))
-        stack_2 = self.bn2(self.conv_2(stack_1))  # next nonlin after sum
-        if self.n_pad_chans != 0:
-            zeros_for_padding = x.new_zeros(
-                (x.shape[0], self.n_pad_chans // 2, x.shape[2], x.shape[3])
-            )
-            x = torch.cat((zeros_for_padding, x, zeros_for_padding), dim=1)
-        out = self.nonlinearity(x + stack_2)
-        return out