Add comments to transformer modules

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