diff --git a/src/Flux.jl b/src/Flux.jl index b7d27406b0..5578e94f08 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -16,7 +16,7 @@ export gradient # Pirate error to catch a common mistake. (Internal function `base` because overloading `update!` is more likely to give ambiguities.) Optimisers.base(dx::Zygote.Grads) = error("Optimisers.jl cannot be used with Zygote.jl's implicit gradients, `Params` & `Grads`") -export Chain, Dense, Maxout, SkipConnection, Parallel, PairwiseFusion, +export Chain, Dense, Embedding, Maxout, SkipConnection, Parallel, PairwiseFusion, RNN, LSTM, GRU, GRUv3, SamePad, Conv, CrossCor, ConvTranspose, DepthwiseConv, AdaptiveMaxPool, AdaptiveMeanPool, GlobalMaxPool, GlobalMeanPool, MaxPool, MeanPool,