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

remove merge conflicts

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-28 15:14:12 +05:30
commit 89bca2d98d
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/optimise/train.jl
View File

@ -1,5 +1,6 @@
using Juno using Juno
using Flux.Tracker: back! using Flux.Tracker: back!
import Base.depwarn
runall(f) = f runall(f) = f
runall(fs::AbstractVector) = () -> foreach(call, fs) runall(fs::AbstractVector) = () -> foreach(call, fs)
@ -59,8 +60,10 @@ function train!(loss, data, opt; cb = () -> ())
l = loss(d...) l = loss(d...)
@interrupts back!(l) @interrupts back!(l)
opt() opt()
cb() == :stop && break if cb() == :stop
@deprecate :stop Flux.stop() depwarn("Use of `:stop` is deprecated; use `Flux.stop()` instead", :stop)
break
end
catch ex catch ex
if ex isa StopException if ex isa StopException
break break