Skip to content

Commit 21749fa

Browse files
committed
fix naming
1 parent 1190149 commit 21749fa

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

src/rules.jl

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -164,20 +164,18 @@ init(o::Rprop, x::AbstractArray) = (zero(x), onevalue(o.eta, x))
164164

165165
function apply!(o::Rprop, state, x, dx)
166166
ℓ, Γ = o.ell, o.gamma
167-
g, η = state
167+
g, η = state
168168

169-
signs = g₀ .* dx
169+
signs = sign.(g .* dx)
170170
signs[signs .> 0] .= ℓ[2]
171171
signs[signs .< 0] .= ℓ[1]
172172
signs[signs .== 0] .= one(eltype(signs))
173173

174-
η₁ = clamp.(η₀ .* signs, Γ[1], Γ[2])
174+
@.. η = clamp* signs, Γ[1], Γ[2])
175+
@.. g = (signs !== ℓ[1]) * dx
175176

176-
@.. g = (signs == ℓ[1]) * dx
177-
178-
dx′ = @lazy η₁ * sign(g₁)
179-
180-
return (g₁, η₁), dx′
177+
dx′ = @lazy η * sign(g)
178+
return (g, η), dx′
181179
end
182180

183181
"""

0 commit comments

Comments
 (0)