Fixed bug in create_gta_features where enumeated over int

scarlettlin2022 · Jul 29, 2019 · 54bb541 · 54bb541
1 parent c549267
commit 54bb541
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_tacotron.py b/train_tacotron.py
@@ -171,7 +171,7 @@ def create_gta_features(model: Tacotron, train_set, save_path: Path):
 
         gta = gta.cpu().numpy()
 
-        for j, item_id in enumerate(len(ids)):
+        for j, item_id in enumerate(ids):
             mel = gta[j][:, :mel_lens[j]]
             mel = (mel + 4) / 8
             np.save(save_path/f'{item_id}.npy', mel, allow_pickle=False)