init durations = None

2026-09-10 11:50:20 +08:00 · 2021-06-22 13:12:52 +02:00
parent 2e31659dd9
commit c09622459e
1 changed files with 1 additions and 0 deletions
@@ -351,6 +351,7 @@ class TrainerTTS:
            speaker_ids = None

        # compute durations from attention masks
+        durations = None
        if attn_mask is not None:
            durations = torch.zeros(attn_mask.shape[0], attn_mask.shape[2])
            for idx, am in enumerate(attn_mask):