diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-07 22:59:57 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-07 22:59:57 +0200 |
commit | ecf050a59ac42c9cf7d12dab44dd51a341c916f2 (patch) | |
tree | 077d3e90d1bc25805112ce5aa234e965dceddec5 /text_recognizer/networks/transformer | |
parent | 87ea75c803400ca74c4f1b863c496165ed802fc2 (diff) |
Removed docs pipeline in noxfile, reformatted code w black
Diffstat (limited to 'text_recognizer/networks/transformer')
-rw-r--r-- | text_recognizer/networks/transformer/layers.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/text_recognizer/networks/transformer/layers.py b/text_recognizer/networks/transformer/layers.py index a44a525..1266d32 100644 --- a/text_recognizer/networks/transformer/layers.py +++ b/text_recognizer/networks/transformer/layers.py @@ -47,11 +47,7 @@ class AttentionLayers(nn.Module): return "a", "f" def _build_network( - self, - causal: bool, - attn_fn: partial, - norm_fn: partial, - ff_fn: partial, + self, causal: bool, attn_fn: partial, norm_fn: partial, ff_fn: partial, ) -> nn.ModuleList: """Configures transformer network.""" layers = nn.ModuleList([]) |