Spaces:
Running
Running
update
Browse files
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 =
|
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(
|