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

fixed docstring and not exporting stop

Browse Source
This commit is contained in:
Dhairya Gandhi 2018-08-22 00:29:07 +05:30
parent ed044e2df7
commit 3d11322d37
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/Flux.jl
View File

@ -21,7 +21,8 @@ include("optimise/Optimise.jl")
using .Optimise using .Optimise
using .Optimise: @epochs using .Optimise: @epochs
export SGD, ADAM, ADAMW, AdaMax, Momentum, Nesterov, export SGD, ADAM, ADAMW, AdaMax, Momentum, Nesterov,
RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM, stop RMSProp, ADAGrad, ADADelta, AMSGrad, NADAM
include("utils.jl") include("utils.jl")
include("onehot.jl") include("onehot.jl")

4
src/optimise/train.jl
View File

@ -18,7 +18,7 @@ struct StopException <: Exception end
""" """
stop() stop()
Call `stop()` in a callback to indicate when a callback condition is met. Call `Flux.stop()` in a callback to indicate when a callback condition is met.
This would trigger the train loop to stop and exit. This would trigger the train loop to stop and exit.
```julia ```julia
@ -89,4 +89,4 @@ macro epochs(n, ex)
@info "Epoch $i" @info "Epoch $i"
$(esc(ex)) $(esc(ex))
end) end)
end end