make location attention optional and keep all attention weights in attention class

2026-09-09 11:16:00 +08:00 · 2019-04-29 11:37:01 +02:00
parent 3ea34c6488
commit e2439fde9a
3 changed files with 62 additions and 40 deletions
@@ -263,5 +263,6 @@ def setup_model(num_chars, c):
            attn_norm=c.attention_norm,
            prenet_type=c.prenet_type,
            forward_attn=c.use_forward_attn,
-            trans_agent=c.transition_agent)
+            trans_agent=c.transition_agent,
+            location_attn=c.location_attn)
    return model