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