diff --git a/src/layers/Layers.jl b/src/layers/Layers.jl index b777b91..b08e4fc 100644 --- a/src/layers/Layers.jl +++ b/src/layers/Layers.jl @@ -11,7 +11,7 @@ using Static: Static using ForwardDiff: ForwardDiff -using Lux: Lux, LuxOps, StatefulLuxLayer, WeightInitializers +using Lux: Lux, LuxOps, StatefulLuxLayer using LuxCore: LuxCore, AbstractLuxLayer, AbstractLuxContainerLayer, AbstractLuxWrapperLayer using MLDataDevices: get_device, CPUDevice using NNlib: NNlib