Merge pull request #485 from dhairyagandhi96/master

Add call back

src/optimise/train.jl

@@ -11,6 +11,7 @@ function update!(opt, xs)
 end
 
 # Callback niceties
+call(f, xs...) = f(xs...)
 runall(f) = f
 runall(fs::AbstractVector) = () -> foreach(call, fs)
 

test/optimise.jl

@@ -1,4 +1,5 @@
 using Flux.Optimise
+using Flux.Optimise: runall
 using Flux.Tracker
 using Test
 @testset "Optimise" begin
@@ -50,4 +51,11 @@ end
               cb = Flux.throttle(() -> (i > 3 && Flux.stop()), 1))
 
   @test 3 < i < 50
+
+  # Test multiple callbacks
+  x = 0
+  fs = [() -> (), () -> x = 1]
+  cbs = runall(fs)
+  cbs()
+  @test x == 1
 end