From 58bb72625cb0b31cc157d87839f8eff3acf0e54b Mon Sep 17 00:00:00 2001 From: Sebastian Callh Date: Wed, 8 Apr 2020 07:53:28 +0200 Subject: [PATCH] Explicitly import `Flux.Optimiser.apply!` in optimiser docs --- docs/src/training/optimisers.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/src/training/optimisers.md b/docs/src/training/optimisers.md index 37288b5d..a5b995ab 100644 --- a/docs/src/training/optimisers.md +++ b/docs/src/training/optimisers.md @@ -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.Optimiser.apply!(o::Momentum, x, Δ) η, ρ = o.eta, o.rho v = get!(o.velocity, x, zero(x))::typeof(x) @. v = ρ * v - η * Δ