Skip to content
Snippets Groups Projects
Commit 8d2097ab authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Remove OCR Manager class

parent 1dcb6c01
No related branches found
No related tags found
1 merge request!244Remove OCR Manager class
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment