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

bug fix for rwd discriminator

This commit is contained in:
erogol 2020-06-05 13:25:53 +02:00
parent 189227c741
commit c5c3f6094c
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vocoder/train.py
View File

@ -327,8 +327,12 @@ def evaluate(model_G, criterion_G, model_D, ap, global_step, epoch):
y_hat = model_G.pqmf_synthesis(y_hat) y_hat = model_G.pqmf_synthesis(y_hat)
y_G_sub = model_G.pqmf_analysis(y_G) y_G_sub = model_G.pqmf_analysis(y_G)
if len(signature(model_D.forward).parameters) == 2:
D_out_fake = model_D(y_hat, c_G)
else:
D_out_fake = model_D(y_hat) D_out_fake = model_D(y_hat)
D_out_real = None D_out_real = None
if c.use_feat_match_loss: if c.use_feat_match_loss:
with torch.no_grad(): with torch.no_grad():
D_out_real = model_D(y_G) D_out_real = model_D(y_G)