From ef63f80644a61b5722b7369d21d1dc93504fe6f7 Mon Sep 17 00:00:00 2001 From: thebhatman Date: Mon, 10 Jun 2019 18:24:18 +0530 Subject: [PATCH] No ops defined for param and data --- src/Flux.jl | 4 ++-- src/layers/basic.jl | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) 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