From 9c94b0c5c060699d57e286137b3556ab55dff840 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 1 Jun 2021 14:53:57 +0200
Subject: [PATCH] init `durations = None`

---
 TTS/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TTS/trainer.py b/TTS/trainer.py
index 34d73874..d81132cf 100644
--- 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):