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 | |
| parent | 87ea75c803400ca74c4f1b863c496165ed802fc2 (diff) | |
Removed docs pipeline in noxfile, reformatted code w black
Diffstat (limited to 'text_recognizer')
| -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([]) |