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

returning

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-18 08:24:49 +05:30
parent 999b00b64d
commit 65a5ecccd2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/optimise/train.jl
View File

@ -37,9 +37,9 @@ function train!(loss, data, opt; cb = () -> ())
cb = try cb = try
runall(cb) runall(cb)
catch e catch e
if e isa StopException || rethrow() if e isa StopException || rethrow(e)
@info "Stop Condition Met" @info "Stop Condition Met"
break :stop
opt = runall(opt) opt = runall(opt)
@progress for d in data @progress for d in data
l = loss(d...) l = loss(d...)