Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Implementation of ATCnet #429

Merged
merged 17 commits into from Nov 25, 2022
Merged

Implementation of ATCnet #429

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
7efc346
Implement ATCNet
cedricrommel Nov 8, 2022
7eb3149
Create ATCNet test
cedricrommel Nov 14, 2022
fcfddc2
Update api.rst and whats_new.rst
cedricrommel Nov 14, 2022
d360176
Modifications to get closer to tensorflow implem
cedricrommel Nov 15, 2022
ca502a6
Typo
cedricrommel Nov 15, 2022
fed334e
Flake8
cedricrommel Nov 15, 2022
b451c62
Fix attention dimension
cedricrommel Nov 17, 2022
825e3b9
Fix concat default 2 averaging as in official code
cedricrommel Nov 17, 2022
567a948
Add test on number of params
cedricrommel Nov 17, 2022
face784
Use home-made multi-head attention implementation
cedricrommel Nov 18, 2022
efb8a74
Add back forgotten dropout in attention block
cedricrommel Nov 21, 2022
b1f7aa9
Fix typo in docstring
cedricrommel Nov 25, 2022
eba3d4b
Add comment about inplace weights modification
cedricrommel Nov 25, 2022
a5ac70e
Move general modules into models.modules
cedricrommel Nov 25, 2022
e010ab2
Avoid wrapping in CausalConv1d layer
cedricrommel Nov 25, 2022
0ff41bb
Avoid wrapping in MaxNormLinear
cedricrommel Nov 25, 2022
6002542
Fix CausalConv1d after wrap removal
cedricrommel Nov 25, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1 change: 1 addition & 0 deletions braindecode/models/__init__.py
View file
Open in desktop
Expand Up @@ -10,6 +10,7 @@
from .eeginception import EEGInception
from .eeginception_erp import EEGInceptionERP
from .eeginception_mi import EEGInceptionMI
from .atcnet import ATCNet
from .tcn import TCN
from .sleep_stager_chambon_2018 import SleepStagerChambon2018
from .sleep_stager_blanco_2020 import SleepStagerBlanco2020
Expand Down