TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

No ops defined for param and data

Browse Source
This commit is contained in:
thebhatman 2019-06-10 18:24:18 +05:30
parent 0ddb5f0265
commit ef63f80644
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Flux.jl
View File

@ -9,8 +9,8 @@ using MacroTools: @forward
using Zygote: Params, @adjoint, gradient using Zygote: Params, @adjoint, gradient
export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, ConvTranspose, MaxPool, MeanPool, export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, ConvTranspose, MaxPool, MeanPool,
DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm, DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm,
params, mapleaves, cpu, gpu, f32, f64 params, mapleaves, cpu, gpu, f32, f64, param, data
include("optimise/Optimise.jl") include("optimise/Optimise.jl")
using .Optimise using .Optimise

3
src/layers/basic.jl
View File

@ -189,3 +189,6 @@ end
function (mo::Maxout)(input::AbstractArray) function (mo::Maxout)(input::AbstractArray)
mapreduce(f -> f(input), (acc, out) -> max.(acc, out), mo.over) mapreduce(f -> f(input), (acc, out) -> max.(acc, out), mo.over)
end end
param(x) = x
data(x) = x