Add init_from_config to GAN

This commit is contained in:
Eren Gölge committed 2022-02-25 10:48:02 +01:00
1 parent 353f913efc
commit e1b4c4ca43
1 file changed
+4
+4
View File
@@ -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)