returning :stop
This commit is contained in:
parent
65a5ecccd2
commit
887bfad312
@ -39,7 +39,7 @@ function train!(loss, data, opt; cb = () -> ())
|
||||
catch e
|
||||
if e isa StopException || rethrow(e)
|
||||
@info "Stop Condition Met"
|
||||
:stop
|
||||
return :stop
|
||||
opt = runall(opt)
|
||||
@progress for d in data
|
||||
l = loss(d...)
|
||||
|
Loading…
Reference in New Issue
Block a user