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

fixes #343

Browse Source
This commit is contained in:
Mike J Innes 2018-09-04 13:37:54 +01:00
parent 53be49b102
commit 93c4a6b4b5
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
REQUIRE
View File

@ -4,7 +4,7 @@ MacroTools 0.3.3
NNlib NNlib
Requires Requires
Adapt Adapt
GZip CodecZlib
Colors Colors
ZipFile ZipFile
AbstractTrees AbstractTrees

10
src/data/mnist.jl
View File

@ -1,11 +1,17 @@
module MNIST module MNIST
using GZip, Colors using CodecZlib, Colors
const Gray = Colors.Gray{Colors.N0f8} const Gray = Colors.Gray{Colors.N0f8}
const dir = joinpath(@__DIR__, "../../deps/mnist") const dir = joinpath(@__DIR__, "../../deps/mnist")
function gzopen(f, file)
open(file) do io
f(GzipDecompressorStream(io))
end
end
function load() function load()
mkpath(dir) mkpath(dir)
cd(dir) do cd(dir) do
@ -17,7 +23,7 @@ function load()
@info "Downloading MNIST dataset" @info "Downloading MNIST dataset"
download("https://cache.julialang.org/http://yann.lecun.com/exdb/mnist/$file.gz", "$file.gz") download("https://cache.julialang.org/http://yann.lecun.com/exdb/mnist/$file.gz", "$file.gz")
open(file, "w") do io open(file, "w") do io
write(io, GZip.open(read, "$file.gz")) write(io, gzopen(read, "$file.gz"))
end end
end end
end end