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

re-export gradient

Browse Source
This commit is contained in:
Mike Innes 2019-07-11 13:55:12 +01:00
parent cf5bc801d3
commit c2cd7dab91
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/Flux.jl
View File

@ -7,10 +7,11 @@ using MacroTools, Juno, Requires, Reexport, Statistics, Random
using MacroTools: @forward using MacroTools: @forward
@reexport using NNlib @reexport using NNlib
using Zygote: Params, @adjoint, gradient using Zygote: Params, @adjoint, gradient
export gradient
export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxPool, MeanPool, export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxPool, MeanPool,
DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm, DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm,
SkipConnection,params, mapleaves, cpu, gpu, f32, f64, param, data SkipConnection, params, mapleaves, cpu, gpu, f32, f64, param, data
include("optimise/Optimise.jl") include("optimise/Optimise.jl")
using .Optimise using .Optimise