Fixed bug where current_step in training loop not updated

scarlettlin2022 · Aug 12, 2019 · 4e940e1 · 4e940e1
1 parent 1b4aaa7
commit 4e940e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_tacotron.py b/train_tacotron.py
@@ -63,10 +63,10 @@ def main():
     optimizer = optim.Adam(model.parameters())
     restore_checkpoint(paths, model, optimizer, create_if_missing=True)
 
-    current_step = model.get_step()
-
     if not force_gta:
         for i, session in enumerate(hp.tts_schedule):
+            current_step = model.get_step()
+
             r, lr, max_step, batch_size = session
 
             training_steps = max_step - current_step