Up the batch size in MNIST example
This commit is contained in:
parent
d550232458
commit
06f22db69a
@ -13,7 +13,7 @@ import numpy as np
|
|||||||
Train a simple deep NN on the MNIST dataset.
|
Train a simple deep NN on the MNIST dataset.
|
||||||
'''
|
'''
|
||||||
|
|
||||||
batch_size = 64
|
batch_size = 128
|
||||||
nb_classes = 10
|
nb_classes = 10
|
||||||
nb_epoch = 20
|
nb_epoch = 20
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user