restrict train! special casing

2020-02-27 20:49:05 +01:00 · 2020-02-27 20:49:05 +01:00 · 487002878e
parent b6c79b38b4
commit 487002878e
1 changed files with 1 additions and 1 deletions
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@ -79,7 +79,7 @@ function train!(loss, ps, data, opt; cb = () -> ())
  cb = runall(cb)
  @progress for d in data
    try
-      if d isa AbstractArray
+      if d isa AbstractArray{<:Number}
        gs = gradient(ps) do
          loss(d)
        end