From bc69a54f44f066589b460a5e8def4a57acb921db Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Fri, 12 Jun 2020 11:54:54 +0200
Subject: [PATCH] visualization bug fix

---
 vocoder/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vocoder/train.py b/vocoder/train.py
index 4d1b1029..326bd90e 100644
--- a/vocoder/train.py
+++ b/vocoder/train.py
@@ -120,8 +120,8 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
         # PQMF formatting
         if y_hat.shape[1] > 1:
             y_hat_sub = y_hat
-            y_hat_vis = y_hat
             y_hat = model_G.pqmf_synthesis(y_hat)
+            y_hat_vis = y_hat
             y_G_sub = model_G.pqmf_analysis(y_G)
 
         if global_step > c.steps_to_start_discriminator: