You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
1.2 KiB
23 lines
1.2 KiB
import torch
|
|
from paddlespeech.t2s.modules.losses import MultiScaleSTFTLoss, MultiMelSpectrogramLoss
|
|
from paddleaudio.audiotools.core.audio_signal import AudioSignal
|
|
|
|
def test_dac_losses():
|
|
for i in range(10):
|
|
loss_origin = torch.load(f'tests/unit/tts/data/{i}-loss.pt')
|
|
recons = AudioSignal(f'tests/unit/tts/data/{i}-recons.wav')
|
|
signal = AudioSignal(f'tests/unit/tts/data/{i}-signal.wav')
|
|
loss_fn_1 = MultiScaleSTFTLoss()
|
|
loss_fn_2 = MultiMelSpectrogramLoss(n_mels=[5, 10, 20, 40, 80, 160, 320], window_lengths=[32, 64, 128, 256, 512, 1024, 2048], mag_weight=0.0, pow=1.0, mel_fmin=[0, 0, 0, 0, 0, 0, 0], mel_fmax=[None, None, None, None, None, None, None])
|
|
#
|
|
# Test AudioSignal
|
|
#
|
|
assert abs(loss_fn_1(recons, signal).item() - loss_origin['stft/loss'].item()) < 1e-5
|
|
assert abs(loss_fn_2(recons, signal).item() - loss_origin['mel/loss'].item()) < 1e-5
|
|
|
|
#
|
|
# Test Tensor
|
|
#
|
|
assert abs(loss_fn_1(recons.audio_data, signal.audio_data).item() - loss_origin['stft/loss'].item()) < 1e-3
|
|
assert abs(loss_fn_2(recons.audio_data, signal.audio_data).item() - loss_origin['mel/loss'].item()) < 1e-3
|