remove hard code

anantzoid · Nov 19, 2016 · 3c5e972 · 3c5e972
1 parent 8a9f0eb
commit 3c5e972
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -70,7 +70,7 @@ def train(conf, data):
         conf.img_height = 28
         conf.img_width = 28
         conf.channel = 1
-        conf.num_batches = 10#data.train.num_examples // conf.batch_size
+        conf.num_batches = data.train.num_examples // conf.batch_size
     else:
         from keras.datasets import cifar10
         data = cifar10.load_data()