From 94e38c05b84dfc283e863f1d96ca5d48be3d3c68 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 27 Sep 2017 18:33:23 +0100 Subject: [PATCH] more informative --- src/optimise/train.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/optimise/train.jl b/src/optimise/train.jl index 8ad437db..0a91e978 100644 --- a/src/optimise/train.jl +++ b/src/optimise/train.jl @@ -8,8 +8,8 @@ function train!(m, data, opt; cb = () -> ()) cb = tocb(cb) @progress for x in data l = m(x...) - isinf(l.data[]) && error("Inf") - isnan(l.data[]) && error("NaN") + isinf(l.data[]) && error("Loss is Inf") + isnan(l.data[]) && error("Loss is NaN") back!(l) opt() cb()