HoneyTian commited on
Commit
4927a3a
·
1 Parent(s): a725e3f
toolbox/torchaudio/losses/spectral.py CHANGED
@@ -293,7 +293,7 @@ class MultiResolutionSTFTLoss(torch.nn.Module):
293
  if not len(fft_size_list) == len(win_size_list) == len(hop_size_list):
294
  raise AssertionError
295
 
296
- loss_fn_list = list()
297
  for n_fft, win_size, hop_size in zip(fft_size_list, win_size_list, hop_size_list):
298
  loss_fn_list.append(
299
  STFTLoss(
 
293
  if not len(fft_size_list) == len(win_size_list) == len(hop_size_list):
294
  raise AssertionError
295
 
296
+ loss_fn_list = nn.ModuleList([])
297
  for n_fft, win_size, hop_size in zip(fft_size_list, win_size_list, hop_size_list):
298
  loss_fn_list.append(
299
  STFTLoss(