diff --git a/src/Flux.jl b/src/Flux.jl index 027bef46..bc7a3a48 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -13,7 +13,7 @@ export Chain, Dense, RNN, LSTM, GRU, Conv2D, param, params, mapleaves using NNlib -export σ, sigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax +export σ, sigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax, conv2d, maxpool2d, avgpool2d include("tracker/Tracker.jl")