Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove select size bug

Browse Source
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
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
y_hat = model_G(c_G)
y_hat = model_G(c_G)[:, :, :y_G.size(2)]
y_hat_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:
# discriminator pass
with torch.no_grad():
y_hat = model_G(c_G)
y_hat = model_G(c_G)[:, :, :y_G.size(2)]
# PQMF formatting
if y_hat.shape[1] > 1: