Add init_from_config to GAN

2026-09-09 11:16:00 +08:00 · 2022-02-25 10:48:02 +01:00
parent 353f913efc
commit e1b4c4ca43
1 changed files with 4 additions and 0 deletions
@@ -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)