Load a language model and decode with it during evaluation

Based on !346 (merged)

Closes #252 (closed)

dan/ocr/manager/training.py

@@ -21,7 +21,8 @@ from torch.nn.parallel import DistributedDataParallel as DDP
 from torch.utils.tensorboard import SummaryWriter
 from tqdm import tqdm
 
-from dan.ocr.decoder import CTCLanguageDecoder
+from dan.ocr.decoder import CTCLanguageDecoder, GlobalHTADecoder
+from dan.ocr.encoder import FCN_Encoder
 from dan.ocr.manager.metrics import Inference, MetricManager
 from dan.ocr.manager.ocr import OCRDatasetManager
 from dan.ocr.mlflow import MLFLOW_AVAILABLE, logging_metrics, logging_tags_metrics
@@ -74,11 +75,11 @@ class GenericTrainingManager:
         self.load_dataset()
 
     @property
-    def encoder(self):
+    def encoder(self) -> FCN_Encoder | None:
         return self.models.get(MODEL_NAME_ENCODER)
 
     @property
-    def decoder(self):
+    def decoder(self) -> GlobalHTADecoder | None:
         return self.models.get(MODEL_NAME_DECODER)
 
     def init_paths(self):