this should be progress'd too

2017-05-01 12:46:02 +01:00 · 2017-05-01 12:46:02 +01:00 · dba6bef245
commit dba6bef245
parent 38852964f6
1 changed files with 1 additions and 1 deletions
--- a/src/training.jl
+++ b/src/training.jl
@ -13,7 +13,7 @@ end
 function train!(m, train, test = [];
                epoch = 1, η = 0.1, loss = mse)
    i = 0
-    for e in 1:epoch
+    @progress for e in 1:epoch
      info("Epoch $e")
      @progress for (x, y) in train
        x, y = tobatch.((x, y))