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

Merge commit '9a7e6e9c'

Browse Source
This commit is contained in:
Mike J Innes 2018-05-01 12:19:05 +01:00
commit 3870027c91
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/Flux.jl
View File

@ -7,22 +7,23 @@ module Flux
using Juno, Requires, Reexport using Juno, Requires, Reexport
using MacroTools: @forward using MacroTools: @forward
export Chain, Dense, RNN, LSTM, GRU, Conv, Conv2D, export Chain, Dense, RNN, LSTM, GRU, Conv,
Dropout, LayerNorm, BatchNorm, Dropout, LayerNorm, BatchNorm,
SGD, ADAM, Momentum, Nesterov, AMSGrad, params, mapleaves, cpu, gpu
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
include("optimise/Optimise.jl") include("optimise/Optimise.jl")
using .Optimise using .Optimise
using .Optimise: @epochs using .Optimise: @epochs
export SGD, ADAM, AdaMax, Momentum, Nesterov,
RMSProp, ADAGrad, ADADelta, AMSGrad
include("utils.jl") include("utils.jl")
include("onehot.jl") include("onehot.jl")

2
src/optimise/Optimise.jl
View File

@ -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}