braindecode 1.2.0.dev184328194__py3-none-any.whl → 1.3.0.dev171478045__py3-none-any.whl

{braindecode-1.2.0.dev184328194.dist-info → braindecode-1.3.0.dev171478045.dist-info}/RECORD

@@ -1,72 +1,75 @@
 braindecode/__init__.py,sha256=Ac3LEEyIHWFY_fFh3eAY1GZUqXcUxVSJwOSUCwGEDvQ,182
 braindecode/classifier.py,sha256=k9vSCtfQbld0YVleDi5rrrmk6k_k5JYEPPBYcNxYjZ8,9807
-braindecode/eegneuralnet.py,sha256=dz8k_-2jV7WqkaX4bQG-dmr-vRT7ZtOwJqomXyC9PTw,15287
+braindecode/eegneuralnet.py,sha256=U6kRdT2u8A2Ca0axMTR8IAESBsvgjLMusAbYappKAOk,15368
 braindecode/regressor.py,sha256=VLfrpiXklwI4onkwue3QmzlBWcvspu0tlrLo9RT1Oiw,9375
 braindecode/util.py,sha256=J-tBcDJNlMTIFW2mfOy6Ko0nsgdP4obRoEVDeg2rFH0,12686
-braindecode/version.py,sha256=Adl2q0noMgIED1dlngWz_nvDbzU6GpgOYSGTS9Fs6io,35
+braindecode/version.py,sha256=g5gWQg7AVLR1mAmDgIYLLB-iuLq1Y2n9miSSXRiLYNY,35
 braindecode/augmentation/__init__.py,sha256=LG7ONqCufYAF9NZt8POIp10lYXb8iSueYkF-CWGK2Ls,1001
 braindecode/augmentation/base.py,sha256=gg7wYsVfa9jfqBddtE03B5ZrPHFFmPl2sa3LOrRnGfo,7325
-braindecode/augmentation/functional.py,sha256=ygkMNEFHaUdRQfk7meMML19FnM406Uf34h-ztKXdJwM,37978
-braindecode/augmentation/transforms.py,sha256=QgLoX6MFaiBH8WoVBgB8eY4x9jZNPMvj20zlwUM8AOs,44245
+braindecode/augmentation/functional.py,sha256=lPhGpZcVtgfQ3oV6p6IQLBCWM_Psa60TwxH3Wj1WyOQ,41133
+braindecode/augmentation/transforms.py,sha256=Ur05yLdROm5pfKTsS2opCWI--X6JwWjP7YMa2KTTZTw,44243
 braindecode/datasets/__init__.py,sha256=CTl8ucbG948ZJqntEBELb-Pn8GsZLfFZLgVcB-fhw4k,891
-braindecode/datasets/base.py,sha256=ED8RQWusMyWf0T7b_HXwouR2Ax47qppEc506AlSzBt0,32155
+braindecode/datasets/base.py,sha256=_qUuMripcBrc04R7j5wOqW41myo5Y_Ku3OqKa4uRqx4,32176
 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
-braindecode/datasets/sleep_physio_challe_18.py,sha256=KTvUtuarOOYu6PHN6H1vcy4W9xilwtZE08n7JSrk8Cs,15414
+braindecode/datasets/sleep_physio_challe_18.py,sha256=66A86_9VssszKrVXowb0oFyL3xbF1VRqQK5FtW33QlM,15427
 braindecode/datasets/sleep_physionet.py,sha256=jieRx6u-MQ4jn_5Zox_pVV8WjBwXKLv9uq4GXRAZ_58,4087
 braindecode/datasets/tuh.py,sha256=iG1hOtdevzKGEVpeuRFDBOnsW_rWa5zEmMFJfYR1hqg,22867
 braindecode/datasets/xy.py,sha256=xT-nS_5jpuVKJ0SGqc7Ia0FVpqj86UfuzcYQdEGZdp0,2986
 braindecode/datautil/__init__.py,sha256=GB9xOudUhJGDyG08PBrnotw6HnWoWIXAHfRNFO-pxSk,1797
-braindecode/datautil/serialization.py,sha256=gLIm9bcuR-XfVdII-RTplUWFRms9qVvVZ0-M6gTucNc,13028
+braindecode/datautil/serialization.py,sha256=g_EVg3oTieqFRattw9OdwMaYjfjANVG-uCS3xVkuHjg,13293
 braindecode/datautil/util.py,sha256=ZfDoxLieKsgI8xcWQqebV-vJ5pJYRvRRHkEwhwpgoKU,674
 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=PhDJl6nBChButabjsmLz_heRcGFCCMKoeUt7k7neNzs,24483
-braindecode/models/attentionbasenet.py,sha256=1uwrtsdEGiBwokkO8A_2SR5zapOTQUBZd4q7hIpR0cw,23359
-braindecode/models/base.py,sha256=9icrWNZBGbh_VLyB9m8g_K1QyK7s3mh8X-hJ29gEbWs,10802
-braindecode/models/biot.py,sha256=T4PymX3penMJcrdfb5Nq6B3P-jyP2laAIu_R9o3uCXo,17512
+braindecode/models/__init__.py,sha256=ovF_WX8ZkXEkleRwYsMMS7ldLPh8_2NzTeYGVqH9ilg,2581
+braindecode/models/atcnet.py,sha256=H2IWMscm3IM4PH8DA_iLkUaeMXgA120DmVld4jBFOCM,32242
+braindecode/models/attentionbasenet.py,sha256=_bml0Ofy7yB12X19a026EYkcLuzZIab0v3sQTqZ5HGQ,30485
+braindecode/models/attn_sleep.py,sha256=m6sdFfD4en2hHf_TpotLPC1hVweJcYZvjgf12bV5FZg,17822
+braindecode/models/base.py,sha256=KjsHVQDdUCAJB4nS-a6eze-H7ayvU4565tsFUcDVxVQ,20212
+braindecode/models/bendr.py,sha256=MZQdYFERVeBJnynEXDlCLdn_I0mJtgzzFuMhCXkbMkg,21591
+braindecode/models/biot.py,sha256=LpJ8tXqQL2Zh_vcQnpUHEpAGQrPHtn2cBSTUPFCW8jQ,17546
 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=_Y0SXprBD74zD8nKPcS9HQ6PoWzfpu-VCY7Tj6R7Xrs,11612
-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=1ZAG0KLDedkodDfqgnGGsoZj6iuU55kGmBlyQo1b47w,16284
-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=i5HzBKTd82fTlKDfB42uc14HpDYxN29SGPfCa4ON5gk,13686
+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
 braindecode/models/hybrid.py,sha256=hA8jwD3_3LL71BxUjRM1dkhqlHU9E9hjuDokh-jBq-4,4024
 braindecode/models/ifnet.py,sha256=Y2bwfko3SDjD74AzgUEzgMhKJFGCCw_Q_Noh5VONEjQ,15137
-braindecode/models/labram.py,sha256=vcrpwiu4F-djtIPscFbtP2Y0jTosyR_cXnOMQQRGPLw,41798
+braindecode/models/labram.py,sha256=1BVGJpPNqXtM_XQ__20p1_9KWVCgvxJ1ICU4HqZW3d8,46600
 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/signal_jepa.py,sha256=UeSkeAM3Qmx8bbAqHCj5nP-PtZM00_5SGA8ibo9mptc,37079
+braindecode/models/patchedtransformer.py,sha256=9TY9l2X4EoCuE9IoOObjubKFRdmsN5lbrVQLnmr66VY,23444
+braindecode/models/sccnet.py,sha256=C7vdwIR5cI6wJCl5f8TnGQG6qinq21y4HG6l-D5AwbY,11971
+braindecode/models/shallow_fbcsp.py,sha256=7U07DJBrm2JHV8v5ja-xuE5-IH5tfmryhJtrfO1n4jk,7531
+braindecode/models/signal_jepa.py,sha256=bBujhM9ItIJisKvbxEi5e1yuV-0mBb41GlyMeEs_TkA,41124
 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/sstdpn.py,sha256=wJv-UYP1q8cMGp2wU1efzIZiigRmkJ8uY22rNB2D7Wc,35077
+braindecode/models/summary.csv,sha256=vFmhpCGFZlxC9Zm8KLBaGRHvZZfdRY85NAGj1Wyv1yU,7209
 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=nrYBdd0FTCoYxgg21oz1UlW-PACx-0-_EyvMQua0QI8,5414
 braindecode/modules/__init__.py,sha256=PD2LpeSHWW_MgEef7-G8ief5gheGObzsIoacchxWuyA,1756
 braindecode/modules/activation.py,sha256=lTO2IjZWBDeXZ4ZVDgLmTDmxHdqyAny3Fsy07HY9tmQ,1466
 braindecode/modules/attention.py,sha256=ISE11jXAvMqKpawZilg8i7lDX5mkuvpEplrh_CtGEkk,24102
@@ -81,7 +84,7 @@ braindecode/modules/util.py,sha256=tVXEhzeTsYrr_wZ5CiXaq3VYGtC5TmGEEW2hMYjTQAE,2
 braindecode/modules/wrapper.py,sha256=Z-aZ4wxA0psYefMOfj03r7D1XjD4az6GpZpaQoDPJv0,2421
 braindecode/preprocessing/__init__.py,sha256=V0iwdzb6DzpUaCabA7I6HmOqXK_XvTbpP5HaEduSJ4s,776
 braindecode/preprocessing/mne_preprocess.py,sha256=_Jczaitqbx16utsUOhnonEcoExf6jPsWNwVOVvoKFfU,2210
-braindecode/preprocessing/preprocess.py,sha256=-9IKjb0THq36m54TK-YRzV18wIkxmVgTcGO2sEH6q98,17665
+braindecode/preprocessing/preprocess.py,sha256=da_-Tn1NLPunsZC2-uzzgCYgdm_Xj-CIJjwf_CTMuFs,17899
 braindecode/preprocessing/windowers.py,sha256=6w6mOnroGWnV7tS23UagZZepswaxaL00S45Jr5AViRE,36551
 braindecode/samplers/__init__.py,sha256=TLuO6gXv2WioJdX671MI_CHVSsOfbjnly1Xv9K3_WdA,452
 braindecode/samplers/base.py,sha256=z_Txp9cEwUmIBL0J6FPJbx1cMSsU9l9mxymRCGqNss0,15111
@@ -93,9 +96,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.dev184328194.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
-braindecode-1.2.0.dev184328194.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
-braindecode-1.2.0.dev184328194.dist-info/METADATA,sha256=PgPq5CmBC6TDByTBtGn3Gtf6yaAJW96CZ_3J5BgGhDc,6883
-braindecode-1.2.0.dev184328194.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-braindecode-1.2.0.dev184328194.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
-braindecode-1.2.0.dev184328194.dist-info/RECORD,,
+braindecode-1.3.0.dev171478045.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
+braindecode-1.3.0.dev171478045.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
+braindecode-1.3.0.dev171478045.dist-info/METADATA,sha256=mwyspEBAyIpNfEvNLbm8yY2DePcO_EuJFLYuVI5dePk,7215
+braindecode-1.3.0.dev171478045.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+braindecode-1.3.0.dev171478045.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
+braindecode-1.3.0.dev171478045.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