um

2016-05-10 19:01:38 +01:00 · 2016-05-10 19:01:38 +01:00 · 109170df22
commit 109170df22
parent 8314da4207
1 changed files with 3 additions and 5 deletions
--- a/examples/MNIST.jl
+++ b/examples/MNIST.jl
@ -7,9 +7,7 @@ const test = data[50_001:60_000]

 const m = Sequence(
  Input(784),
-  Dense(30),
-  Sigmoid(),
-  Dense(10),
-  Sigmoid())
+  Dense(30), Sigmoid(),
+  Dense(10), Sigmoid())

-Flux.train!(m, epoch = 30)
+@time Flux.train!(m, train, test, epoch = 30)