From 8100360c71b1cc0a327f9a32e7edd169e7186ad5 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 20 Feb 2019 17:14:08 +0100 Subject: [PATCH] bug fix for initialization of the model --- train.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index fd257d31..9c3b2b8f 100644 --- a/train.py +++ b/train.py @@ -357,7 +357,12 @@ def evaluate(model, criterion, criterion_st, ap, current_step): def main(args): num_chars = len(phonemes) if c.use_phonemes else len(symbols) - model = Tacotron(num_chars, c.embedding_size, ap.num_freq, ap.num_mels, c.r, c.memory_size) + model = Tacotron(num_chars=num_chars, + embedding_dim=c.embedding_size, + linear_dim=ap.num_freq, + mel_dim=ap.num_mels, + r=c.r, + memory_size=c.memory_size) print(" | > Num output units : {}".format(ap.num_freq), flush=True) optimizer = optim.Adam(model.parameters(), lr=c.lr, weight_decay=0)