bug fix for initialization of the model

2026-09-10 11:50:20 +08:00 · 2019-02-20 17:14:08 +01:00
parent 71f1f0c4c0
commit 8100360c71
1 changed files with 6 additions and 1 deletions
@@ -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)