diff --git a/src/Flux.jl b/src/Flux.jl index a4f8cd93..361fadfd 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -9,8 +9,8 @@ using MacroTools: @forward using Zygote: Params, @adjoint, gradient export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, ConvTranspose, MaxPool, MeanPool, - DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm, - params, mapleaves, cpu, gpu, f32, f64 + DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm, + params, mapleaves, cpu, gpu, f32, f64, param, data include("optimise/Optimise.jl") using .Optimise diff --git a/src/layers/basic.jl b/src/layers/basic.jl index dea0089f..a86b9310 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -189,3 +189,6 @@ end function (mo::Maxout)(input::AbstractArray) mapreduce(f -> f(input), (acc, out) -> max.(acc, out), mo.over) end + +param(x) = x +data(x) = x