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

Remove select size bug

This commit is contained in:
rishikksh20 2021-03-05 15:41:58 +05:30 committed by Eren Gölge
parent b533474e3b
commit e656e8b108
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/bin/train_vocoder_gan.py
View File

@ -321,7 +321,7 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
############################## ##############################
# generator pass # generator pass
y_hat = model_G(c_G) y_hat = model_G(c_G)[:, :, :y_G.size(2)]
y_hat_sub = None y_hat_sub = None
y_G_sub = None y_G_sub = None
@ -373,7 +373,7 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
if global_step >= c.steps_to_start_discriminator: if global_step >= c.steps_to_start_discriminator:
# discriminator pass # discriminator pass
with torch.no_grad(): with torch.no_grad():
y_hat = model_G(c_G) y_hat = model_G(c_G)[:, :, :y_G.size(2)]
# PQMF formatting # PQMF formatting
if y_hat.shape[1] > 1: if y_hat.shape[1] > 1: