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()