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))