Changesat windowing and some comments

2026-09-10 11:50:20 +08:00 · 2019-04-12 16:13:40 +02:00
parent 455667d2a4
commit 3c2d500f53
2 changed files with 21 additions and 14 deletions
@@ -252,12 +252,14 @@ def setup_model(num_chars, c):
        model = MyModel(
            num_chars=num_chars,
            r=c.r,
+            attn_win=c.windowing,
            attn_norm=c.attention_norm,
            memory_size=c.memory_size)
    elif c.model.lower() == "tacotron2":
        model = MyModel(
            num_chars=num_chars,
            r=c.r,
+            attn_win=c.windowing,
            attn_norm=c.attention_norm,
            prenet_type=c.prenet_type,
            forward_attn=c.use_forward_attn,