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

bug fix

Browse Source
This commit is contained in:
erogol
2020-05-31 14:18:29 +02:00
parent d5bbcd0487
commit 29915ba85c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vocoder/train.py
+2 -2
View File
@@ -137,7 +137,7 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
if global_step > c.steps_to_start_discriminator:
# run D with or without cond. features
if len(signature(model_D).parameters) == 2:
if len(signature(model_D.forward).parameters) == 2:
D_out_fake = model_D(in_fake_D, c_G)
else:
D_out_fake = model_D(in_fake_D)
@@ -195,7 +195,7 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
optimizer_D.zero_grad()
# run D with or without cond. features
if len(signature(model_D).parameters) == 2:
if len(signature(model_D.forward).parameters) == 2:
D_out_fake = model_D(y_hat.detach(), c_D)
D_out_real = model_D(y_D, c_D)
else: