From 8af1e9c12e22f28186739664febb6b0285e3e212 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Seungwon=20Park=20=EB=B0=95=EC=8A=B9=EC=9B=90?=
 <yyyyy@snu.ac.kr>
Date: Mon, 11 Nov 2019 10:40:55 +0900
Subject: [PATCH] fix #22 (#28)

* fix wrong implementation of generator

* update torch hub model url
---
 hubconf.py         | 2 +-
 model/generator.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/hubconf.py b/hubconf.py
index 27d3e96..f8a76a6 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -5,7 +5,7 @@
 model_params = {
     'nvidia_tacotron2_LJ11_epoch3200': {
         'mel_channel': 80,
-        'model_url': 'https://github.com/seungwonpark/melgan/releases/download/v0.1-alpha/nvidia_tacotron2_LJ11_epoch3200.pt',
+        'model_url': 'https://github.com/seungwonpark/melgan/releases/download/v0.2-alpha/nvidia_tacotron2_LJ11_epoch3200_v02.pt',
     },
 }
 
diff --git a/model/generator.py b/model/generator.py
index 2614aee..5be2797 100644
--- a/model/generator.py
+++ b/model/generator.py
@@ -37,7 +37,7 @@ def __init__(self, mel_channel):
             ResStack(32),
 
             nn.LeakyReLU(),
-            nn.utils.weight_norm(nn.ConvTranspose1d(32, 1, kernel_size=7, stride=1, padding=3)),
+            nn.utils.weight_norm(nn.Conv1d(32, 1, kernel_size=7, stride=1, padding=3)),
             nn.Tanh(),
         )