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

Use @info for logging

Browse Source
This commit is contained in:
Josh Christie 2018-08-11 14:42:33 +01:00
parent 710a65fe72
commit c8307a0627
5 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/data/mnist.jl
View File

@ -14,7 +14,7 @@ function load()
"t10k-images-idx3-ubyte", "t10k-images-idx3-ubyte",
"t10k-labels-idx1-ubyte"] "t10k-labels-idx1-ubyte"]
isfile(file) && continue isfile(file) && continue
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, GZip.open(read, "$file.gz"))

2
src/data/sentiment.jl
View File

@ -5,7 +5,7 @@ using ..Data: deps
function load() function load()
isfile(deps("sentiment.zip")) || return isfile(deps("sentiment.zip")) || return
info("Downloading sentiment treebank dataset") @info "Downloading sentiment treebank dataset"
download("https://cache.julialang.org/https://nlp.stanford.edu/sentiment/trainDevTestTrees_PTB.zip", download("https://cache.julialang.org/https://nlp.stanford.edu/sentiment/trainDevTestTrees_PTB.zip",
deps("sentiment.zip")) deps("sentiment.zip"))
end end

2
src/optimise/train.jl
View File

@ -59,7 +59,7 @@ hello
""" """
macro epochs(n, ex) macro epochs(n, ex)
:(@progress for i = 1:$(esc(n)) :(@progress for i = 1:$(esc(n))
info("Epoch $i") @info "Epoch $i"
$(esc(ex)) $(esc(ex))
end) end)
end end

2
test/cuda/cuda.jl
View File

@ -1,7 +1,7 @@
using Flux, Flux.Tracker, CuArrays, Test using Flux, Flux.Tracker, CuArrays, Test
using Flux: gpu using Flux: gpu
info("Testing Flux/GPU") @info "Testing Flux/GPU"
@testset "CuArrays" begin @testset "CuArrays" begin

2
test/cuda/cudnn.jl
View File

@ -1,6 +1,6 @@
using Flux, CuArrays, Test using Flux, CuArrays, Test
info("Testing Flux/CUDNN") @info "Testing Flux/CUDNN"
@testset "RNN" begin @testset "RNN" begin
@testset for R in [RNN, GRU, LSTM] @testset for R in [RNN, GRU, LSTM]