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

fix second unsqueeze

Browse Source
This commit is contained in:
Mike J Innes 2017-04-27 15:04:48 +01:00
parent f4acd286c8
commit f5f5e77239
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/MNIST.jl
View File

@ -19,4 +19,4 @@ model(unsqueeze(data[1][1]))
Flux.train!(model, train, test, η = 1e-4) Flux.train!(model, train, test, η = 1e-4)
# An example prediction post-training # An example prediction post-training
model(data[1][1]) model(unsqueeze(data[1][1]))