diff options
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/models/base.py | 1 | ||||
| -rw-r--r-- | text_recognizer/networks/transformer/decoder_block.py | 2 | ||||
| -rw-r--r-- | text_recognizer/networks/transformer/norm.py | 2 |
3 files changed, 2 insertions, 3 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index 4587a30..4dd5cdf 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -62,7 +62,6 @@ class LitBase(LightningModule): ) monitor = self.lr_scheduler_config.pop("monitor") interval = self.lr_scheduler_config.pop("interval") - return { "monitor": monitor, "interval": interval, 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 |