From 3c5e97285d7118cb62e3e5ec48799b1df911fd36 Mon Sep 17 00:00:00 2001 From: Anant Gupta Date: Sat, 19 Nov 2016 21:49:40 +0530 Subject: [PATCH] remove hard code --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 1305231..8cad634 100644 --- 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()