From dba6bef24588d8b21ccafa50fbef47cd037e401e Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Mon, 1 May 2017 12:46:02 +0100
Subject: [PATCH] this should be progress'd too

---
 src/training.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/training.jl b/src/training.jl
index 4ead5076..b0daf211 100644
--- 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))