Fix #813 - GlowTTS training (#814)

* Fix #813 * Update glow_tts recipe * Fix glow-tts test * Linter fix * Run data dep init only in training
2026-09-12 12:11:47 +08:00 · 2021-09-17 20:06:55 +02:00
parent f563415052
commit 2766dd1d6e
7 changed files with 114 additions and 51 deletions
@@ -63,7 +63,7 @@ class GlowTTSTrainTest(unittest.TestCase):
            optimizer.zero_grad()
            outputs = model.forward(input_dummy, input_lengths, mel_spec, mel_lengths, None)
            loss_dict = criterion(
-                outputs["model_outputs"],
+                outputs["z"],
                outputs["y_mean"],
                outputs["y_log_scale"],
                outputs["logdet"],