diff --git a/examples/bench_cifar10.py b/examples/bench_cifar10.py index f8bf95d..2b52fec 100644 --- a/examples/bench_cifar10.py +++ b/examples/bench_cifar10.py @@ -56,6 +56,7 @@ def load(name): mlp.Layer("Softmax", units=n_targets)], n_iter=50, n_stable=10, + batch_size=25, learning_rate=0.002, learning_rule="momentum", valid_size=0.1,