From 1d282ea8aa69f73ae49f6ae24c9bdbff5f52f656 Mon Sep 17 00:00:00 2001
From: Dominic Monn <monn.dominic@gmail.com>
Date: Thu, 13 Dec 2018 09:37:53 -0800
Subject: [PATCH] fixed dropout/keep_prob confusion

---
 nets/densenet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nets/densenet.py b/nets/densenet.py
index 3b5fa30..6494232 100644
--- a/nets/densenet.py
+++ b/nets/densenet.py
@@ -46,7 +46,8 @@ def _conv(inputs, num_filters, kernel_size, stride=1, dropout_rate=None,
     net = slim.conv2d(net, num_filters, kernel_size)
 
     if dropout_rate:
-      net = tf.nn.dropout(net)
+        keep_prob = 1.0 - dropout_rate
+        net = slim.dropout(net, keep_prob=keep_prob)
 
     net = slim.utils.collect_named_outputs(outputs_collections, sc.name, net)