Add init_from_config to GAN

2021-11-24 12:59:51 +01:00 · 2021-11-24 12:59:51 +01:00 · e1b4c4ca43
parent 353f913efc
commit e1b4c4ca43
1 changed files with 4 additions and 0 deletions
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@ -361,3 +361,7 @@ class GAN(BaseVocoder):
    def get_criterion(self):
        """Return criterions for the optimizers"""
        return [GeneratorLoss(self.config), DiscriminatorLoss(self.config)]
+
+    @staticmethod
+    def init_from_config(config: Coqpit) -> "GAN":
+        return GAN(config)