Skip to content

Commit 6011c67

Browse files
authored
add dropout
1 parent b12b51d commit 6011c67

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

CNNs/SqueezeNet.py

+2
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ def __init__(self, inputs, nb_classes=1000, is_training=True):
3737
name="maxpool8")
3838
# fire9
3939
net = self._fire(net, 64, 256, "fire9")
40+
# dropout
41+
net = tf.layers.dropout(net, 0.5, training=is_training)
4042
# conv10
4143
net = tf.layers.conv2d(net, 1000, [1, 1], strides=[1, 1],
4244
padding="SAME", activation=tf.nn.relu,

0 commit comments

Comments
 (0)