From 9b83e665fcef1c2ef8928ec81f06d7cd74e0863d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 24 Nov 2021 18:45:32 +0100 Subject: [PATCH] Add init_from_config as an abstract class --- TTS/model.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/TTS/model.py b/TTS/model.py index a7c64dde..efa00b2a 100644 --- a/TTS/model.py +++ b/TTS/model.py @@ -130,6 +130,15 @@ class BaseModel(nn.Module, ABC): """ ... + @staticmethod + @abstractmethod + def init_from_config(config: Coqpit): + """Init the model from given config. + + Override this depending on your model. + """ + pass + def get_optimizer(self) -> Union["Optimizer", List["Optimizer"]]: """Setup an return optimizer or optimizers.""" pass @@ -150,3 +159,4 @@ class BaseModel(nn.Module, ABC): def format_batch(self): pass +