AbstractArray -> DenseArray for params!()

FluxML · Aug 24, 2022 · 6718314 · 6718314
1 parent a2fee8e
commit 6718314
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/functor.jl b/src/functor.jl
@@ -36,7 +36,7 @@ Possible values include:
 """
 trainmode!(m, mode = true) = mode isa Bool ? testmode!(m, !mode) : testmode!(m, mode)
 
-params!(p::Params, x::AbstractArray{<:Number}, seen = IdSet()) = Functors.isleaf(x) && push!(p, x)
+params!(p::Params, x::DenseArray{<:Number}, seen = IdSet()) = Functors.isleaf(x) && push!(p, x)
 
 function params!(p::Params, x, seen = IdSet())
   x in seen && return