mirror of https://github.com/coqui-ai/TTS.git
Add init_from_config as an abstract class
This commit is contained in:
parent
3eca5ad060
commit
9b83e665fc
10
TTS/model.py
10
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"]]:
|
def get_optimizer(self) -> Union["Optimizer", List["Optimizer"]]:
|
||||||
"""Setup an return optimizer or optimizers."""
|
"""Setup an return optimizer or optimizers."""
|
||||||
pass
|
pass
|
||||||
|
@ -150,3 +159,4 @@ class BaseModel(nn.Module, ABC):
|
||||||
|
|
||||||
def format_batch(self):
|
def format_batch(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue