show accuracy less frequently

This commit is contained in:
Mike J Innes 2016-08-25 23:12:16 +01:00
parent de48c1d589
commit 821efb0cf6
1 changed files with 2 additions and 1 deletions

View File

@ -17,7 +17,8 @@ function train!(m, train, test = []; epoch = 1, batch = 10, η = 0.1)
any(isnan, pred) && error("NaN")
err = mse!(, pred, y)
back!(m, , x)
i % batch == 0 && (update!(m, η); @show accuracy(m, test))
i % batch == 0 && update!(m, η)
i % 1000 == 0 && @show accuracy(m, test)
end
end
return m