Made epochs 1-indexed

scarlettlin2022 · Aug 13, 2019 · 6b79dbb · 6b79dbb
1 parent 4e940e1
commit 6b79dbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_tacotron.py b/train_tacotron.py
@@ -116,7 +116,7 @@ def tts_train_loop(paths: Paths, model: Tacotron, optimizer, train_set, lr, trai
     total_iters = len(train_set)
     epochs = train_steps // total_iters + 1
 
-    for e in range(epochs):
+    for e in range(1, epochs+1):
 
         start = time.time()
         running_loss = 0