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

Callback being called now

Browse Source
This commit is contained in:
thebhatman 2019-06-20 00:37:54 +05:30
parent e6d5846e49
commit b194e7e3a8
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/optimise/train.jl
View File

@ -67,6 +67,7 @@ function train!(loss, ps, data, opt; cb = () -> ())
loss(d...) loss(d...)
end end
update!(opt, ps, gs) update!(opt, ps, gs)
cb()
catch ex catch ex
if ex isa StopException if ex isa StopException
break break