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

use Juno progress bar

Browse Source
This commit is contained in:
Mike J Innes 2016-09-06 18:37:39 +01:00
parent e262221eeb
commit 62ede8cd80
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Flux.jl
View File

@ -1,6 +1,6 @@
module Flux module Flux
using MacroTools, Lazy, Flow using MacroTools, Lazy, Flow, Juno
# Zero Flux Given # Zero Flux Given

2
src/utils.jl
View File

@ -11,7 +11,7 @@ function train!(m, train, test = []; epoch = 1, batch = 10, η = 0.1)
i = 0 i = 0
∇ = zeros(length(train[1][2])) ∇ = zeros(length(train[1][2]))
for _ in 1:epoch for _ in 1:epoch
for (x, y) in train @progress for (x, y) in train
i += 1 i += 1
pred = m(x) pred = m(x)
any(isnan, pred) && error("NaN") any(isnan, pred) && error("NaN")