From c2cd7dab9126dff5401a58bb7ed3dbbbd9427ecd Mon Sep 17 00:00:00 2001 From: Mike Innes Date: Thu, 11 Jul 2019 13:55:12 +0100 Subject: [PATCH] re-export gradient --- src/Flux.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/Flux.jl b/src/Flux.jl index d3537b9e..2a5fb3b5 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -7,10 +7,11 @@ using MacroTools, Juno, Requires, Reexport, Statistics, Random using MacroTools: @forward @reexport using NNlib using Zygote: Params, @adjoint, gradient +export gradient export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxPool, MeanPool, 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") using .Optimise