From 8c7768e8d321efec558e12bff9b89b2de615d541 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Thu, 13 May 2021 23:02:20 +0200 Subject: Decoder module working --- text_recognizer/networks/transformer/attention.py | 1 + 1 file changed, 1 insertion(+) (limited to 'text_recognizer/networks/transformer/attention.py') diff --git a/text_recognizer/networks/transformer/attention.py b/text_recognizer/networks/transformer/attention.py index eabeadf..a3b53f0 100644 --- a/text_recognizer/networks/transformer/attention.py +++ b/text_recognizer/networks/transformer/attention.py @@ -23,6 +23,7 @@ class Attention(nn.Module): dropout_rate: float = 0.0, causal: bool = False, ) -> None: + super().__init__() self.scale = dim ** -0.5 self.num_heads = num_heads self.causal = causal -- cgit v1.2.3-70-g09d2