update

mindee · Apr 11, 2024 · 21c5005 · 21c5005
1 parent 3c46181
commit 21c5005
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/doctr/models/recognition/crnn/tensorflow.py b/doctr/models/recognition/crnn/tensorflow.py
@@ -136,15 +136,13 @@ def __init__(
         top_paths: int = 1,
         cfg: Optional[Dict[str, Any]] = None,
     ) -> None:
-        # Initialize kernels
-        h, w, c = feature_extractor.output_shape[1:]
-
         super().__init__()
+        self.feat_extractor = feature_extractor
+        h, w, c = feature_extractor.output_shape[1:]
         self.vocab = vocab
         self.max_length = w
         self.cfg = cfg
         self.exportable = exportable
-        self.feat_extractor = feature_extractor
 
         self.decoder = Sequential([
             layers.Bidirectional(layers.LSTM(units=rnn_units, return_sequences=True)),