[example] simplify mnist simple

deepmusic · Aug 5, 2017 · 053e0fd · 053e0fd
1 parent a0172b2
commit 053e0fd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/example/tutorial_mnist_simple.py b/example/tutorial_mnist_simple.py
@@ -18,11 +18,9 @@
 # define the network
 network = tl.layers.InputLayer(x, name='input')
 network = tl.layers.DropoutLayer(network, keep=0.8, name='drop1')
-network = tl.layers.DenseLayer(network, n_units=800,
-                                act = tf.nn.relu, name='relu1')
+network = tl.layers.DenseLayer(network, 800, tf.nn.relu, name='relu1')
 network = tl.layers.DropoutLayer(network, keep=0.5, name='drop2')
-network = tl.layers.DenseLayer(network, n_units=800,
-                                act = tf.nn.relu, name='relu2')
+network = tl.layers.DenseLayer(network, 800, tf.nn.relu, name='relu2')
 network = tl.layers.DropoutLayer(network, keep=0.5, name='drop3')
 # the softmax is implemented internally in tl.cost.cross_entropy(y, y_) to
 # speed up computation, so we use identity here.