summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/layers.py
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks/transformer/layers.py')
-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([])