update

examples/conv_tasnet/step_2_train_model.py

@@ -206,7 +206,7 @@ def main():
     neg_stoi_loss_fn = NegSTOILoss(sample_rate=8000, reduction="mean").to(device)
     lds_loss_fn = LSDLoss(reduction="mean").to(device)
     mr_stft_loss_fn = MultiResolutionSTFTLoss(
-        fft_size_list=[256, 512, 1024],
+        # fft_size_list=[256, 512, 1024],
         win_size_list=[120, 240, 480],
         hop_size_list=[25, 50, 100],
         reduction="mean"
@@ -240,7 +240,7 @@ def main():
         total_lds_loss = 0.
         total_batches = 0.
         progress_bar = tqdm(
-            desc="Training; epoch: {}".format(idx_epoch),
+            desc="Training; epoch-{}".format(idx_epoch),
         )
         for batch in train_data_loader:
             clean_audios, noisy_audios = batch
@@ -305,7 +305,7 @@ def main():
         total_batches = 0.
 
         progress_bar = tqdm(
-            desc="Evaluation; epoch: {}".format(idx_epoch),
+            desc="Evaluation; epoch-{}".format(idx_epoch),
         )
         with torch.no_grad():
             for batch in valid_data_loader: