fix typo (coqui-ai#2475)

jstaerk · Apr 3, 2023 · 95fa2c9 · 95fa2c9
1 parent 91cf1b2
commit 95fa2c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py
@@ -200,7 +200,7 @@ class for each corresponding step.
         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 @@ def __init__(self, c, ga_sigma=0.4):
             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()