From 95fa2c9fd6f6d3f517dbd1b67f4ca0f036b64c50 Mon Sep 17 00:00:00 2001 From: Yingzhi WANG <41187612+BenoitWang@users.noreply.github.com> Date: Mon, 3 Apr 2023 23:31:09 +0200 Subject: [PATCH] fix typo (#2475) --- TTS/tts/layers/losses.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py index 98be5b88..e12abf20 100644 --- a/TTS/tts/layers/losses.py +++ b/TTS/tts/layers/losses.py @@ -200,7 +200,7 @@ class BCELossMasked(nn.Module): return loss -class DifferentailSpectralLoss(nn.Module): +class DifferentialSpectralLoss(nn.Module): """Differential Spectral Loss https://arxiv.org/ftp/arxiv/papers/1909/1909.10302.pdf""" @@ -335,7 +335,7 @@ class TacotronLoss(torch.nn.Module): self.criterion_ga = GuidedAttentionLoss(sigma=ga_sigma) # differential spectral loss if c.postnet_diff_spec_alpha > 0 or c.decoder_diff_spec_alpha > 0: - self.criterion_diff_spec = DifferentailSpectralLoss(loss_func=self.criterion) + self.criterion_diff_spec = DifferentialSpectralLoss(loss_func=self.criterion) # ssim loss if c.postnet_ssim_alpha > 0 or c.decoder_ssim_alpha > 0: self.criterion_ssim = SSIMLoss()