From 63b328142ad6b73c144d3f700cd87c4c429596cc Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 27 Apr 2017 17:43:38 +0100 Subject: [PATCH] print epochs again --- src/utils.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/utils.jl b/src/utils.jl index 5f1e2807..35f7343d 100644 --- 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