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

2
src/optimise/train.jl
View File

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