use Juno progress bar

2016-09-06 18:37:39 +01:00 · 2016-09-06 18:37:39 +01:00 · 62ede8cd80
commit 62ede8cd80
parent e262221eeb
2 changed files with 2 additions and 2 deletions
--- a/src/Flux.jl
+++ b/src/Flux.jl
@ -1,6 +1,6 @@
 module Flux

-using MacroTools, Lazy, Flow
+using MacroTools, Lazy, Flow, Juno

 # Zero Flux Given

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