init `durations = None`

2021-06-01 14:53:57 +02:00 · 2021-06-01 14:53:57 +02:00 · c09622459e
parent 2e31659dd9
commit c09622459e
1 changed files with 1 additions and 0 deletions
--- a/TTS/trainer.py
+++ b/TTS/trainer.py
@ -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):