fix dropout

strubell · Jun 18, 2018 · 3ae0f92 · 3ae0f92
1 parent 94f9dcf
commit 3ae0f92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/models/parsers/parser.py b/lib/models/parsers/parser.py
@@ -78,7 +78,7 @@ def __call__(self, dataset, moving_params=None):
       predicate_embed_inputs = vocabs[4].embedding_lookup(inputs[:, :, 3], moving_params=self.moving_params)
       embed_inputs = tf.concat([embed_inputs, predicate_embed_inputs], axis=2)
 
-    top_recur = tf.nn.dropout(embed_inputs, self.input_dropout)
+    top_recur = tf.nn.dropout(embed_inputs, self.input_dropout if self.moving_params is None else 1.0)
 
     attn_weights_by_layer = {}