summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/decoder.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-10-02 02:58:00 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-10-02 02:58:00 +0200
commit75c41a3866c7f842996f6f046b94563be4132a83 (patch)
tree9861c6f737d3331ef423363ae3bd974593328e6b /text_recognizer/networks/transformer/decoder.py
parent09bb189ba19304d26ede208fe43c3c882c309d7f (diff)
Update docstrings
Diffstat (limited to 'text_recognizer/networks/transformer/decoder.py')
-rw-r--r--text_recognizer/networks/transformer/decoder.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/transformer/decoder.py b/text_recognizer/networks/transformer/decoder.py
index 09d2dce..826bc13 100644
--- a/text_recognizer/networks/transformer/decoder.py
+++ b/text_recognizer/networks/transformer/decoder.py
@@ -30,7 +30,7 @@ class Decoder(nn.Module):
context: Optional[Tensor] = None,
mask: Optional[Tensor] = None,
) -> Tensor:
- """Applies the network to the signals."""
+ """Applies attention blocks."""
for block in self.blocks:
x = block(
x=x,