diff --git a/src/functor.jl b/src/functor.jl index 73483ab9..1d3e1bb2 100644 --- a/src/functor.jl +++ b/src/functor.jl @@ -39,7 +39,7 @@ end trainable(m) = functor(m)[1] -params!(p::Params, x::AbstractArray{<:Real}, seen = IdSet()) = push!(p, x) +params!(p::Params, x::AbstractArray{<:Number}, seen = IdSet()) = push!(p, x) function params!(p::Params, x, seen = IdSet()) x in seen && return