From 8d2097abcfddf7836d4b20869f4601efb726fad8 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Mon, 14 Aug 2023 12:11:00 +0200
Subject: [PATCH] Remove OCR Manager class
---
dan/ocr/manager/training.py | 9 ++-------
1 file changed, 2 insertions(+), 7 deletions(-)
diff --git a/dan/ocr/manager/training.py b/dan/ocr/manager/training.py
index edb906aa..1034bb65 100644
--- a/dan/ocr/manager/training.py
+++ b/dan/ocr/manager/training.py
@@ -903,15 +903,10 @@ class GenericTrainingManager:
return info_dict
-class OCRManager(GenericTrainingManager):
- def __init__(self, params):
- super(OCRManager, self).__init__(params)
- self.params["model"]["vocab_size"] = len(self.dataset.charset)
-
-
-class Manager(OCRManager):
+class Manager(GenericTrainingManager):
def __init__(self, params):
super(Manager, self).__init__(params)
+ self.params["model"]["vocab_size"] = len(self.dataset.charset)
def load_save_info(self, info_dict):
if "curriculum_config" in info_dict:
--
GitLab