diff options
Diffstat (limited to 'text_recognizer/networks/transformer/residual.py')
| -rw-r--r-- | text_recognizer/networks/transformer/residual.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/text_recognizer/networks/transformer/residual.py b/text_recognizer/networks/transformer/residual.py index 1547df6..825a0fc 100644 --- a/text_recognizer/networks/transformer/residual.py +++ b/text_recognizer/networks/transformer/residual.py @@ -3,6 +3,8 @@ from torch import nn, Tensor class Residual(nn.Module): + """Residual block.""" + def forward(self, x: Tensor, residual: Tensor) -> Tensor: """Applies the residual function.""" return x + residual |