diff options
Diffstat (limited to 'text_recognizer/networks/transformer')
| -rw-r--r-- | text_recognizer/networks/transformer/decoder.py | 2 |
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, |