TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

um

Browse Source
This commit is contained in:
Mike J Innes 2016-05-10 19:01:38 +01:00
parent 8314da4207
commit 109170df22
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/MNIST.jl
View File

@ -7,9 +7,7 @@ const test = data[50_001:60_000]
const m = Sequence( const m = Sequence(
Input(784), Input(784),
Dense(30), Dense(30), Sigmoid(),
Sigmoid(), Dense(10), Sigmoid())
Dense(10),
Sigmoid())
Flux.train!(m, epoch = 30) @time Flux.train!(m, train, test, epoch = 30)