Store max_seq_len in BaseDecoder

2022-05-27 14:42:40 +08:00 · 2022-05-27 14:42:40 +08:00 · 00f821315e
parent bbbefaeb31
commit 00f821315e
1 changed files with 1 additions and 0 deletions
--- 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)