diff options
Diffstat (limited to 'text_recognizer/networks/transformer')
| -rw-r--r-- | text_recognizer/networks/transformer/decoder_block.py | 2 | ||||
| -rw-r--r-- | text_recognizer/networks/transformer/norm.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/networks/transformer/decoder_block.py b/text_recognizer/networks/transformer/decoder_block.py index f7ae454..b8eb5c4 100644 --- a/text_recognizer/networks/transformer/decoder_block.py +++ b/text_recognizer/networks/transformer/decoder_block.py @@ -10,7 +10,7 @@ from text_recognizer.networks.transformer.ff import FeedForward class DecoderBlock(nn.Module): - """Decoder block.""" + """Residual decoder block.""" def __init__( self, diff --git a/text_recognizer/networks/transformer/norm.py b/text_recognizer/networks/transformer/norm.py index 0bd2e16..1431327 100644 --- a/text_recognizer/networks/transformer/norm.py +++ b/text_recognizer/networks/transformer/norm.py @@ -4,7 +4,7 @@ Copied from lucidrains: https://github.com/lucidrains/x-transformers/blob/main/x_transformers/x_transformers.py """ -from typing import Dict, Optional, Type +from typing import Optional, Type import torch from torch import Tensor, nn |