Spaces:
Sleeping
Sleeping
File size: 419 Bytes
e613cea |
1 2 3 4 5 6 7 8 9 10 11 12 |
from ._stft_loss import MultiResolutionSTFTLoss
def subband_stft_loss(h, y_mb, y_hat_mb):
sub_stft_loss = MultiResolutionSTFTLoss(
h.train.fft_sizes, h.train.hop_sizes, h.train.win_lengths
)
y_mb = y_mb.view(-1, y_mb.size(2))
y_hat_mb = y_hat_mb.view(-1, y_hat_mb.size(2))
sub_sc_loss, sub_mag_loss = sub_stft_loss(y_hat_mb[:, : y_mb.size(-1)], y_mb)
return sub_sc_loss + sub_mag_loss
|