more exports
This commit is contained in:
parent
ee89a7797e
commit
e186b958dd
16
src/Flux.jl
16
src/Flux.jl
@ -7,26 +7,25 @@ module Flux
|
|||||||
using Juno, Requires, Reexport
|
using Juno, Requires, Reexport
|
||||||
using MacroTools: @forward
|
using MacroTools: @forward
|
||||||
|
|
||||||
export Chain, Dense, RNN, LSTM, GRU, Conv, Conv2D,
|
|
||||||
Dropout, LayerNorm, BatchNorm,
|
|
||||||
SGD, ADAM, Momentum, Nesterov, AMSGrad,
|
|
||||||
param, params, mapleaves, cpu, gpu
|
|
||||||
|
|
||||||
@reexport using NNlib
|
@reexport using NNlib
|
||||||
using NNlib: @fix
|
using NNlib: @fix
|
||||||
|
|
||||||
include("tracker/Tracker.jl")
|
include("tracker/Tracker.jl")
|
||||||
using .Tracker
|
using .Tracker
|
||||||
export Tracker
|
using .Tracker: data
|
||||||
import .Tracker: data
|
export TrackedArray, TrackedVector, TrackedMatrix, param, back!
|
||||||
|
|
||||||
include("optimise/Optimise.jl")
|
include("optimise/Optimise.jl")
|
||||||
using .Optimise
|
using .Optimise
|
||||||
using .Optimise: @epochs
|
using .Optimise: @epochs
|
||||||
|
export train!,
|
||||||
|
SGD, ADAM, Momentum, Nesterov,
|
||||||
|
RMSProp, ADAGrad, ADADelta, AMSGrad
|
||||||
|
|
||||||
include("utils.jl")
|
include("utils.jl")
|
||||||
include("onehot.jl")
|
include("onehot.jl")
|
||||||
include("treelike.jl")
|
include("treelike.jl")
|
||||||
|
export params, mapleaves, cpu, gpu, onehot, batch, glorot_normal, glorot_uniform
|
||||||
|
|
||||||
include("layers/stateless.jl")
|
include("layers/stateless.jl")
|
||||||
include("layers/basic.jl")
|
include("layers/basic.jl")
|
||||||
@ -34,6 +33,9 @@ include("layers/conv.jl")
|
|||||||
include("layers/recurrent.jl")
|
include("layers/recurrent.jl")
|
||||||
include("layers/normalise.jl")
|
include("layers/normalise.jl")
|
||||||
|
|
||||||
|
export Chain, Dense, RNN, LSTM, GRU, Conv2D,
|
||||||
|
Dropout, LayerNorm, BatchNorm
|
||||||
|
|
||||||
include("data/Data.jl")
|
include("data/Data.jl")
|
||||||
|
|
||||||
@require CuArrays include("cuda/cuda.jl")
|
@require CuArrays include("cuda/cuda.jl")
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
module Optimise
|
module Optimise
|
||||||
|
|
||||||
export update!, params, train!,
|
export train!,
|
||||||
SGD, ADAM, AdaMax, Momentum, Nesterov, RMSProp, ADAGrad, ADADelta, AMSGrad
|
SGD, ADAM, AdaMax, Momentum, Nesterov, RMSProp, ADAGrad, ADADelta, AMSGrad
|
||||||
|
|
||||||
struct Param{T}
|
struct Param{T}
|
||||||
|
Loading…
Reference in New Issue
Block a user