From bdda28c77798d3c08913fd9c9059710f288e0e41 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 5 Jun 2022 21:20:16 +0200 Subject: Fix kwargs --- text_recognizer/networks/conformer/scale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/text_recognizer/networks/conformer/scale.py b/text_recognizer/networks/conformer/scale.py index da736a3..d012b81 100644 --- a/text_recognizer/networks/conformer/scale.py +++ b/text_recognizer/networks/conformer/scale.py @@ -9,5 +9,5 @@ class Scale(nn.Module): self.scale = scale self.fn = fn - def forward(self, x: Tensor, kwargs: Dict) -> Tensor: + def forward(self, x: Tensor, **kwargs) -> Tensor: return self.fn(x, **kwargs) * self.scale -- cgit v1.2.3-70-g09d2