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

returning :stop

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-18 08:28:47 +05:30
parent 65a5ecccd2
commit 887bfad312
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/optimise/train.jl
View File

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