more test

FluxML · Oct 9, 2020 · 56fafa6 · 56fafa6
1 parent 7850a88
commit 56fafa6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/optimise.jl b/test/optimise.jl
@@ -55,6 +55,16 @@ end
 
   @test i==0 #all skipped
 
+  Flux.train!(
+              () -> (sleep(0.1); i==8 && Flux.skip(); i+=1),
+              (),
+              Iterators.repeated((), 10),
+              Descent()
+             )
+
+  @test i==8 #skip after i hit 8
+
+  i = 0
   Flux.train!(() -> (sleep(0.1); i += 1; l),
               (),
               Iterators.repeated((), 100),