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

properly importing functions

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-20 13:30:05 +05:30
parent 1228e9c5e2
commit e239eb1105
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Flux.jl
View File

@ -7,7 +7,7 @@ using MacroTools: @forward
export Chain, Dense, RNN, LSTM, GRU, Conv, export Chain, Dense, RNN, LSTM, GRU, Conv,
Dropout, LayerNorm, BatchNorm, Dropout, LayerNorm, BatchNorm,
params, mapleaves, cpu, gpu params, mapleaves, cpu, gpu, stop, StopException
@reexport using NNlib @reexport using NNlib
using NNlib: @fix using NNlib: @fix
@ -24,7 +24,6 @@ export SGD, ADAM, ADAMW, AdaMax, Momentum, Nesterov,
RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM
include("utils.jl") include("utils.jl")
export stop, StopException
include("onehot.jl") include("onehot.jl")
include("treelike.jl") include("treelike.jl")
@ -39,3 +38,4 @@ include("data/Data.jl")
@init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" include("cuda/cuda.jl") @init @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" include("cuda/cuda.jl")
end # module end # module

2
src/optimise/train.jl
View File

@ -1,6 +1,6 @@
using Juno using Juno
using Flux.Tracker: back! using Flux.Tracker: back!
# include("../utils.jl") import Flux: stop, StopException
runall(f) = f runall(f) = f
runall(fs::AbstractVector) = () -> foreach(call, fs) runall(fs::AbstractVector) = () -> foreach(call, fs)