Skip to content

Commit a37cab2

Browse files
committed
fix?
1 parent cfc3e67 commit a37cab2

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/rules.jl

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,11 +166,16 @@ function apply!(o::Rprop, state, x, dx)
166166
ℓ, Γ = o.ell, o.gamma
167167
g, η = state
168168

169-
g, η = broadcast(g, η, dx) do g, η, dx
170-
g * dx > 0 ? (g, min* ℓ[2], Γ[2])) : g * dx < 0 ? (zero(dx), max* ℓ[1], Γ[1])) : (g, η)
169+
η = broadcast(g, η, dx) do g, η, dx
170+
g * dx > 0 ? min* ℓ[2], Γ[2]) : g * dx < 0 ? max* ℓ[1], Γ[1]) : η
171+
end
172+
173+
g = broadcast(g, dx) do g, dx
174+
g * dx < 0 ? zero(dx) : dx
171175
end
172176

173177
dx′ = @lazy η * sign(g)
178+
174179
return (g, η), dx′
175180
end
176181

0 commit comments

Comments
 (0)