fix: device(NN) should only give warning once (#1156)

LuxDL · Jan 2, 2025 · 222e173 · 222e173
1 parent 1a0c480
commit 222e173
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/LuxCore/ext/LuxCoreMLDataDevicesExt.jl b/lib/LuxCore/ext/LuxCoreMLDataDevicesExt.jl
@@ -9,7 +9,7 @@ MLDataDevices.isleaf(::AbstractLuxLayer) = true
 function Adapt.adapt_storage(::AbstractDevice, x::AbstractLuxLayer)
     @warn "Lux layers are stateless and hence don't participate in device transfers. \
            Apply this function on the parameters and states generated using \
-           `LuxCore.setup`."
+           `LuxCore.setup`." maxlog=1
     return x
 end