summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-07 22:59:57 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-07 22:59:57 +0200
commitecf050a59ac42c9cf7d12dab44dd51a341c916f2 (patch)
tree077d3e90d1bc25805112ce5aa234e965dceddec5 /text_recognizer/networks/transformer
parent87ea75c803400ca74c4f1b863c496165ed802fc2 (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.py6
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([])