From 9c98272cf03f91a8f6f6f4ae9fd26cd24df44061 Mon Sep 17 00:00:00 2001 From: Dhairya Gandhi Date: Sun, 19 Aug 2018 17:38:00 +0530 Subject: [PATCH] catching exception --- src/optimise/train.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/optimise/train.jl b/src/optimise/train.jl index 12a1a272..0663b9d4 100644 --- a/src/optimise/train.jl +++ b/src/optimise/train.jl @@ -43,7 +43,7 @@ function train!(loss, data, opt; cb = () -> ()) try cb() catch ex - if ex isa StopException() + if ex isa StopException @info "Stop condition met" break else