setup_loss() in layer/__init__.py

2026-09-09 11:16:00 +08:00 · 2021-06-28 17:03:19 +02:00
parent a20a1c7d06
commit 844abb3b1d
1 changed files with 15 additions and 0 deletions
@@ -0,0 +1,15 @@
+from TTS.tts.layers.losses import *
+
+
+def setup_loss(config):
+    if config.model.lower() in ["tacotron", "tacotron2"]:
+        model = TacotronLoss(config)
+    elif config.model.lower() == "glow_tts":
+        model = GlowTTSLoss()
+    elif config.model.lower() == "speedy_speech":
+        model = SpeedySpeechLoss(config)
+    elif config.model.lower() == "align_tts":
+        model = AlignTTSLoss(config)
+    else:
+        raise ValueError(f" [!] loss for model {config.model.lower()} cannot be found.")
+    return model