Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix typo (#2475)

This commit is contained in:
Yingzhi WANG 2023-04-03 23:31:09 +02:00 committed by GitHub
parent 91cf1b2da9
commit 95fa2c9fd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/layers/losses.py
View File

@ -200,7 +200,7 @@ class BCELossMasked(nn.Module):
return loss return loss
class DifferentailSpectralLoss(nn.Module): class DifferentialSpectralLoss(nn.Module):
"""Differential Spectral Loss """Differential Spectral Loss
https://arxiv.org/ftp/arxiv/papers/1909/1909.10302.pdf""" 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) self.criterion_ga = GuidedAttentionLoss(sigma=ga_sigma)
# differential spectral loss # differential spectral loss
if c.postnet_diff_spec_alpha > 0 or c.decoder_diff_spec_alpha > 0: 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 # ssim loss
if c.postnet_ssim_alpha > 0 or c.decoder_ssim_alpha > 0: if c.postnet_ssim_alpha > 0 or c.decoder_ssim_alpha > 0:
self.criterion_ssim = SSIMLoss() self.criterion_ssim = SSIMLoss()