diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-10-04 22:06:53 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-10-04 22:06:53 +0200 |
| commit | 3ec10be9141bf71fb10d699b31a66b4e5046973c (patch) | |
| tree | 0e0942ae71ea72b62f43c733fc6e54395f10cb7c /text_recognizer/networks/transformer | |
| parent | 8626406b872f7cf4b69a9dcc8bdfb777c6f1300d (diff) | |
Format and rewording docstring
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 |