From e57ae77bbbcb103f8ab614d33d8feb0420b19ede Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Sun, 3 Sep 2017 02:44:32 -0400
Subject: [PATCH] juno progress

---
 src/optimise/train.jl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 763d4c69..af03edb4 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -1,9 +1,10 @@
+using Juno
 using Flux.Tracker: back!
 
 function train!(m, data, opt; epoch = 1, cb = () -> ())
-  for e in 1:epoch
+  @progress for e in 1:epoch
     epoch > 1 && info("Epoch $e")
-    for (x, y) in data
+    @progress for (x, y) in data
       back!(m(x, y))
       opt()
       cb()