Data loader bug fix 2

2026-09-12 12:11:47 +08:00 · 2018-03-26 11:07:15 -07:00
parent 3c084177c6
commit 1ff8d6d2b7
4 changed files with 9 additions and 5 deletions
@@ -82,7 +82,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
        linear_input = data[2]
        mel_input = data[3]
        mel_lengths = data[4]
-
+        
        current_step = num_iter + args.restore_step + epoch * len(data_loader) + 1

        # setup lr