deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove unnecessary f0 loss in discrimnator

Browse Source
This commit is contained in:
lyuxiang.lx
2024-10-18 15:58:19 +08:00
parent a4db3db8ed
commit 5bd5dfecab
3 changed files with 27 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cosyvoice/hifigan/hifigan.py
View File

@@ -64,6 +64,5 @@ class HiFiGan(nn.Module):
loss_tpr = tpr_loss(y_d_rs, y_d_gs, self.tpr_loss_tau)
else:
loss_tpr = torch.zeros(1).to(device)
loss_f0 = F.l1_loss(generated_f0, pitch_feat)
loss = loss_disc + self.tpr_loss_weight * loss_tpr + loss_f0
return {'loss': loss, 'loss_disc': loss_disc, 'loss_tpr': loss_tpr, 'loss_f0': loss_f0}
loss = loss_disc + self.tpr_loss_weight * loss_tpr
return {'loss': loss, 'loss_disc': loss_disc, 'loss_tpr': loss_tpr}