Skip to content

Commit 3c5e972

Browse files
committed
remove hard code
1 parent 8a9f0eb commit 3c5e972

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def train(conf, data):
7070
conf.img_height = 28
7171
conf.img_width = 28
7272
conf.channel = 1
73-
conf.num_batches = 10#data.train.num_examples // conf.batch_size
73+
conf.num_batches = data.train.num_examples // conf.batch_size
7474
else:
7575
from keras.datasets import cifar10
7676
data = cifar10.load_data()

0 commit comments

Comments
 (0)