make dropout oprional #2

This commit is contained in:
Eren Golge
2019-05-12 17:35:31 +02:00
parent 820d18c922
commit 6331bccefc
3 changed files with 6 additions and 5 deletions
+1
View File
@@ -262,6 +262,7 @@ def setup_model(num_chars, c):
attn_win=c.windowing,
attn_norm=c.attention_norm,
prenet_type=c.prenet_type,
prenet_dropout=c.prenet_dropout,
forward_attn=c.use_forward_attn,
trans_agent=c.transition_agent,
location_attn=c.location_attn)