TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

print epochs again

Browse Source
This commit is contained in:
Mike J Innes 2017-04-27 17:43:38 +01:00
parent 1edabe6052
commit 63b328142a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/utils.jl
View File

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