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

this should be progress'd too

Browse Source
This commit is contained in:
Mike J Innes 2017-05-01 12:46:02 +01:00
parent 38852964f6
commit dba6bef245
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/training.jl
View File

@ -13,7 +13,7 @@ end
function train!(m, train, test = []; function train!(m, train, test = [];
epoch = 1, η = 0.1, loss = mse) epoch = 1, η = 0.1, loss = mse)
i = 0 i = 0
for e in 1:epoch @progress for e in 1:epoch
info("Epoch $e") 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))