Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add init_from_config to GAN

This commit is contained in:
Eren Gölge 2021-11-24 12:59:51 +01:00
parent 353f913efc
commit e1b4c4ca43
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/vocoder/models/gan.py
View File

@ -361,3 +361,7 @@ class GAN(BaseVocoder):
def get_criterion(self): def get_criterion(self):
"""Return criterions for the optimizers""" """Return criterions for the optimizers"""
return [GeneratorLoss(self.config), DiscriminatorLoss(self.config)] return [GeneratorLoss(self.config), DiscriminatorLoss(self.config)]
@staticmethod
def init_from_config(config: Coqpit) -> "GAN":
return GAN(config)