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

vocoder train.py update

This commit is contained in:
erogol 2020-07-15 10:55:45 +02:00
parent 35ad3090a1
commit 2c51ffd6dd
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vocoder/train.py
View File

@ -124,6 +124,7 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
y_hat_vis = y_hat y_hat_vis = y_hat
y_G_sub = model_G.pqmf_analysis(y_G) y_G_sub = model_G.pqmf_analysis(y_G)
scores_fake, feats_fake, feats_real = None, None, None
if global_step > c.steps_to_start_discriminator: if global_step > c.steps_to_start_discriminator:
# run D with or without cond. features # run D with or without cond. features
@ -146,8 +147,6 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
_, feats_real = D_out_real _, feats_real = D_out_real
else: else:
scores_fake = D_out_fake scores_fake = D_out_fake
else:
scores_fake, feats_fake, feats_real = None, None, None
# compute losses # compute losses
loss_G_dict = criterion_G(y_hat, y_G, scores_fake, feats_fake, loss_G_dict = criterion_G(y_hat, y_G, scores_fake, feats_fake,
@ -328,6 +327,7 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
y_G_sub = model_G.pqmf_analysis(y_G) y_G_sub = model_G.pqmf_analysis(y_G)
scores_fake, feats_fake, feats_real = None, None, None
if global_step > c.steps_to_start_discriminator: if global_step > c.steps_to_start_discriminator:
if len(signature(model_D.forward).parameters) == 2: if len(signature(model_D.forward).parameters) == 2:
@ -349,8 +349,7 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
_, feats_real = D_out_real _, feats_real = D_out_real
else: else:
scores_fake = D_out_fake scores_fake = D_out_fake
else: feats_fake, feats_real = None, None
scores_fake, feats_fake, feats_real = None, None, None
# compute losses # compute losses
loss_G_dict = criterion_G(y_hat, y_G, scores_fake, feats_fake, loss_G_dict = criterion_G(y_hat, y_G, scores_fake, feats_fake,