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

fixed throwing exception

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-19 14:54:54 +05:30
parent 887bfad312
commit 2aa057ec08
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/optimise/train.jl
View File

@ -36,8 +36,8 @@ Multiple optimisers and callbacks can be passed to `opt` and `cb` as arrays.
function train!(loss, data, opt; cb = () -> ()) function train!(loss, data, opt; cb = () -> ())
cb = try cb = try
runall(cb) runall(cb)
catch e catch ex
if e isa StopException || rethrow(e) if ex isa StopException || rethrow(ex)
@info "Stop Condition Met" @info "Stop Condition Met"
return :stop return :stop
opt = runall(opt) opt = runall(opt)

2
src/utils.jl
View File

@ -157,5 +157,5 @@ end
struct StopException <: Exception end struct StopException <: Exception end
function stop() function stop()
throw(StopException) throw(StopException())
end end