print epochs again

2017-04-27 17:43:38 +01:00 · 2017-04-27 17:43:38 +01:00 · 63b328142a
commit 63b328142a
parent 1edabe6052
1 changed files with 2 additions and 1 deletions
--- a/src/utils.jl
+++ b/src/utils.jl
@ -9,7 +9,8 @@ tobatch(xs) = tobatch(batchone(xs))

 function train!(m, train, test = []; epoch = 1, η = 0.1)
    i = 0
-    for _ in 1:epoch
+    for e in 1:epoch
+      info("Epoch $e")
      @progress for (x, y) in train
        x, y = tobatch.((x, y))
        i += 1