update

examples/clean_unet_aishell/step_2_train_model.py

@@ -184,9 +184,12 @@ def main():
     ae_loss_fn = nn.L1Loss(reduction="mean").to(device)
 
     mr_stft_loss_fn = MultiResolutionSTFTLoss(
-        fft_sizes=[256, 512, 1024],
-        hop_sizes=[25, 50, 120],
-        win_lengths=[120, 240, 600],
+        fft_sizes=[512],
+        hop_sizes=[50],
+        win_lengths=[240],
+        # fft_sizes=[256, 512, 1024],
+        # hop_sizes=[25, 50, 120],
+        # win_lengths=[120, 240, 600],
         # fft_sizes=[512, 1024, 2048],
         # hop_sizes=[50, 120, 240],
         # win_lengths=[240, 600, 1200],