wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-21 12:40:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fix vocoder training

Browse Source
This commit is contained in:
erogol
2020-06-15 19:04:05 +02:00
parent 16eb15a5ff
commit acb367be26
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vocoder/layers/losses.py
+2 -2
View File
@@ -199,8 +199,8 @@ class GeneratorLoss(nn.Module):
self.stft_loss_weight = C.stft_loss_weight
self.subband_stft_loss_weight = C.subband_stft_loss_weight
self.mse_gan_loss_weight = C.mse_gan_loss_weight
self.hinge_gan_loss_weight = C.hinge_gan_loss_weight
self.mse_gan_loss_weight = C.mse_G_loss_weight
self.hinge_gan_loss_weight = C.hinge_G_loss_weight
self.feat_match_loss_weight = C.feat_match_loss_weight
if C.use_stft_loss: