From c3d32c395c18ba9dd29109abc003d0b9a7ac5880 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Fri, 4 Nov 2016 22:28:12 +0000
Subject: [PATCH] new progress integration

---
 src/backend/tensorflow/recurrent.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/tensorflow/recurrent.jl b/src/backend/tensorflow/recurrent.jl
index 85fc7bf2..d77487a7 100644
--- a/src/backend/tensorflow/recurrent.jl
+++ b/src/backend/tensorflow/recurrent.jl
@@ -55,9 +55,9 @@ function Flux.train!(m::SeqModel, Xs, Ys; epoch = 1, η = 0.1,
   Y = placeholder(Float32)
   Loss = loss(Y, output)/batchlen/seqlen
   minimize_op = TensorFlow.train.minimize(opt(), Loss)
-  for e in 1:epoch
+  @progress "training" for e in 1:epoch
     info("Epoch $e\n")
-    @progress for (i, (x, y)) in enumerate(zip(Xs,Ys))
+    @progress "epoch" for (i, (x, y)) in enumerate(zip(Xs,Ys))
       out = run(sess, vcat(outstates..., output, Loss, minimize_op),
                 merge(Dict(input=>batchone(x), Y=>batchone(y)),
                       Dict(zip(instates, state))))