more informative

2017-09-27 18:33:23 +01:00 · 2017-09-27 18:33:23 +01:00 · 94e38c05b8
commit 94e38c05b8
parent 96d1c55263
1 changed files with 2 additions and 2 deletions
--- 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()