move calculating total_batch to outside

phutoan31299 · May 8, 2018 · 905b3d0 · 905b3d0
1 parent 7e4b46c
commit 905b3d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lab-10-5-mnist_nn_dropout.py b/lab-10-5-mnist_nn_dropout.py
@@ -15,6 +15,7 @@
 learning_rate = 0.001
 training_epochs = 15
 batch_size = 100
+total_batch = int(mnist.train.num_examples / batch_size)
 
 # input place holders
 X = tf.placeholder(tf.float32, [None, 784])
@@ -66,7 +67,6 @@
 # train my model
 for epoch in range(training_epochs):
     avg_cost = 0
-    total_batch = int(mnist.train.num_examples / batch_size)
 
     for i in range(total_batch):
         batch_xs, batch_ys = mnist.train.next_batch(batch_size)