fix #22 (#28)

* fix wrong implementation of generator * update torch hub model url
seungwonpark · Nov 11, 2019 · 8af1e9c · 8af1e9c
1 parent 7ac999e
commit 8af1e9c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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(),
         )