Fix params! to work with complex numbers

2019-09-29 12:23:41 +02:00 · 2019-09-29 12:23:41 +02:00 · d91677f651
commit d91677f651
parent e2b93bc78a
1 changed files with 1 additions and 0 deletions
--- 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