From 00f821315ebe4a5afa54dbc95063401b8b0294af Mon Sep 17 00:00:00 2001 From: gaotongxiao Date: Fri, 27 May 2022 14:42:40 +0800 Subject: [PATCH] Store max_seq_len in BaseDecoder --- mmocr/models/textrecog/decoders/base_decoder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmocr/models/textrecog/decoders/base_decoder.py b/mmocr/models/textrecog/decoders/base_decoder.py index 8b26c078..b48f01af 100644 --- a/mmocr/models/textrecog/decoders/base_decoder.py +++ b/mmocr/models/textrecog/decoders/base_decoder.py @@ -42,6 +42,7 @@ class BaseDecoder(BaseModule): f'but got {type(dictionary)}') self.loss = None self.postprocessor = None + self.max_seq_len = max_seq_len if loss is not None: assert isinstance(loss, dict)