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

Remove select size bug

Browse Source
This commit is contained in:
rishikksh20 authored and Eren Gölge committed 2021-04-08 11:20:33 +02:00
1 parent b533474e3b
commit e656e8b108
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
TTS/bin/train_vocoder_gan.py
+2 -2
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: