update

examples/conv_tasnet/step_2_train_model.py

@@ -267,12 +267,10 @@ def main():
             initial=step_idx,
             desc="Training; epoch-{}".format(epoch_idx),
         )
-        for idx, train_batch in enumerate(train_data_loader):
-            if idx < step_idx:
-                continue
+        for train_batch in train_data_loader:
             clean_audios, noisy_audios = train_batch
-            clean_audios = clean_audios.to(device)
-            noisy_audios = noisy_audios.to(device)
+            clean_audios: torch.Tensor = clean_audios.to(device)
+            noisy_audios: torch.Tensor = noisy_audios.to(device)
 
             denoise_audios = model.forward(noisy_audios)
             denoise_audios = torch.squeeze(denoise_audios, dim=1)