From 3ec10be9141bf71fb10d699b31a66b4e5046973c Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 4 Oct 2022 22:06:53 +0200 Subject: Format and rewording docstring --- text_recognizer/networks/transformer/decoder_block.py | 2 +- text_recognizer/networks/transformer/norm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'text_recognizer/networks/transformer') 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 -- cgit v1.2.3-70-g09d2