From 29915ba85c77db944071ea1e0be2b153c2e74102 Mon Sep 17 00:00:00 2001 From: erogol Date: Sun, 31 May 2020 14:18:29 +0200 Subject: [PATCH] bug fix --- vocoder/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vocoder/train.py b/vocoder/train.py index 0d039e73..a30e3bff 100644 --- a/vocoder/train.py +++ b/vocoder/train.py @@ -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: