diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:23:01 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-05 21:23:01 +0200 |
| commit | 7b5d89046e2317045d0534e6d1cdea438b6b2546 (patch) | |
| tree | 1e15db0076c5b2a94df548d0654e447b65b24fa8 /text_recognizer/networks/transformer/norm.py | |
| parent | da79e64c3603db8fd20a45cb419a7a1128c0fe01 (diff) | |
Fix kwargs
Diffstat (limited to 'text_recognizer/networks/transformer/norm.py')
| -rw-r--r-- | text_recognizer/networks/transformer/norm.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/transformer/norm.py b/text_recognizer/networks/transformer/norm.py index be38346..537246d 100644 --- a/text_recognizer/networks/transformer/norm.py +++ b/text_recognizer/networks/transformer/norm.py @@ -34,7 +34,7 @@ class PreNorm(nn.Module): self.norm = nn.LayerNorm(normalized_shape) self.fn = fn - def forward(self, x: Tensor, **kwargs: Dict) -> Tensor: + def forward(self, x: Tensor, **kwargs) -> Tensor: """Applies pre norm.""" x = self.norm(x) return self.fn(x, **kwargs) |