HoneyTian commited on
Commit
78494be
·
1 Parent(s): 73f64b2
examples/clean_unet_aishell/step_2_train_model.py CHANGED
@@ -184,9 +184,12 @@ def main():
184
  ae_loss_fn = nn.L1Loss(reduction="mean").to(device)
185
 
186
  mr_stft_loss_fn = MultiResolutionSTFTLoss(
187
- fft_sizes=[512, 1024, 2048],
188
- hop_sizes=[50, 120, 240],
189
- win_lengths=[240, 600, 1200],
 
 
 
190
  sc_lambda=0.5,
191
  mag_lambda=0.5,
192
  band="full"
 
184
  ae_loss_fn = nn.L1Loss(reduction="mean").to(device)
185
 
186
  mr_stft_loss_fn = MultiResolutionSTFTLoss(
187
+ fft_sizes=[256, 512, 1024],
188
+ hop_sizes=[25, 50, 120],
189
+ win_lengths=[120, 240, 600],
190
+ # fft_sizes=[512, 1024, 2048],
191
+ # hop_sizes=[50, 120, 240],
192
+ # win_lengths=[240, 600, 1200],
193
  sc_lambda=0.5,
194
  mag_lambda=0.5,
195
  band="full"