exclude

LuxDL · Oct 18, 2024 · 7202dbc · 7202dbc
1 parent 44a1dea
commit 7202dbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/public.jl b/src/public.jl
@@ -348,7 +348,7 @@ for (dev) in (:CPU, :CUDA, :AMDGPU, :Metal, :oneAPI, :XLA)
         (D::$(ldev))(x::Union{Tuple, NamedTuple}) = map(D, x)
         function (D::$(ldev))(x)
             isleaf(x) && return Adapt.adapt(D, x)
-            return Functors.fmap(D, x)
+            return Functors.fmap(D, x, exclude=isleaf)
         end
     end
 end