We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b12b51d commit 6011c67Copy full SHA for 6011c67
CNNs/SqueezeNet.py
@@ -37,6 +37,8 @@ def __init__(self, inputs, nb_classes=1000, is_training=True):
37
name="maxpool8")
38
# fire9
39
net = self._fire(net, 64, 256, "fire9")
40
+ # dropout
41
+ net = tf.layers.dropout(net, 0.5, training=is_training)
42
# conv10
43
net = tf.layers.conv2d(net, 1000, [1, 1], strides=[1, 1],
44
padding="SAME", activation=tf.nn.relu,
0 commit comments