diff --git a/src/optimise/train.jl b/src/optimise/train.jl index 92208bc1..0663b9d4 100644 --- a/src/optimise/train.jl +++ b/src/optimise/train.jl @@ -47,6 +47,7 @@ function train!(loss, data, opt; cb = () -> ()) @info "Stop condition met" break else + println(ex) rethrow(ex) end end