update

examples/conv_tasnet/step_2_train_model.py

@@ -267,7 +267,9 @@ def main():
             initial=step_idx,
             desc="Training; epoch-{}".format(epoch_idx),
         )
-        for train_batch in train_data_loader:
+        for idx, train_batch in enumerate(train_data_loader):
+            if idx < step_idx:
+                continue
             clean_audios, noisy_audios = train_batch
             clean_audios = clean_audios.to(device)
             noisy_audios = noisy_audios.to(device)