new progress integration

2016-11-04 22:28:12 +00:00 · 2016-11-04 22:28:12 +00:00 · c3d32c395c
commit c3d32c395c
parent 65531b8689
1 changed files with 2 additions and 2 deletions
--- 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))))