From 3d866cbd08a8a28596d5bca2e21d4db3e8c3abe2 Mon Sep 17 00:00:00 2001 From: Vedant Puri Date: Thu, 2 Jan 2025 19:54:09 +0530 Subject: [PATCH] device(NN) should only give warning once --- lib/LuxCore/ext/LuxCoreMLDataDevicesExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/LuxCore/ext/LuxCoreMLDataDevicesExt.jl b/lib/LuxCore/ext/LuxCoreMLDataDevicesExt.jl index 14c842eb11..a505c37d77 100644 --- 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