From 810d8b2403dd0a229063c5693deac694871243f6 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 27 Oct 2021 22:16:04 +0200
Subject: Add comments to transformer modules

---
 text_recognizer/networks/transformer/residual.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'text_recognizer/networks/transformer/residual.py')

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
-- 
cgit v1.2.3-70-g09d2