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

Merge #1113 

1113: Explicitly import `Flux.Optimiser.apply!` in optimiser docs r=dhairyagandhi96 a=SebastianCallh

Closes #1111 

Co-authored-by: Sebastian Callh <sebastian.callh@peltarion.com>
Co-authored-by: Sebastian Callh <sebastian.callh@gmail.com>
This commit is contained in:
bors[bot] 2020-04-08 08:14:17 +00:00 committed by GitHub
parent 4f693e02cb 7b55bf40a3
commit d6c53c13db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/training/optimisers.md
View File

@ -79,7 +79,7 @@ Momentum(eta::Real, rho::Real) = Momentum(eta, rho, IdDict())
The `Momentum` type will act as our optimiser in this case. Notice that we have added all the parameters as fields, along with the velocity which we will use as our state dictionary. Each parameter in our models will get an entry in there. We can now define the rule applied when this optimiser is invoked.
```julia
function apply!(o::Momentum, x, Δ)
function Flux.Optimise.apply!(o::Momentum, x, Δ)
η, ρ = o.eta, o.rho
v = get!(o.velocity, x, zero(x))::typeof(x)
@. v = ρ * v - η * Δ