Make it actually work

This commit is contained in:
Filippo Vicentini 2019-09-29 12:28:01 +02:00 committed by GitHub
parent d91677f651
commit 14e94c291e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -40,7 +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)
params!(p::Params, x::AbstractArray{<:Complex{<:Real}}, seen = IdSet()) = push!(p, x)
function params!(p::Params, x, seen = IdSet())
x in seen && return