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(), )