fix dropout bug

kuanghuei · Jun 26, 2018 · 4da7957 · 4da7957
1 parent 7e3b466
commit 4da7957
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -127,7 +127,7 @@ def ref_set_encoder(self, v_S, is_training=True, scope="ref_set_encoder"):
                 biases_initializer=tf.zeros_initializer())
 
             if is_training:
-                h = tf.nn.dropout(h0, keep_prob=self.keep_prob)
+                h0 = tf.nn.dropout(h0, keep_prob=self.keep_prob)
 
             h = tf.contrib.layers.fully_connected(inputs=h0, 
                 num_outputs=256,