Skip to content

Commit 3c5cb5a

Browse files
committed
Merge branch 'proximal_entropy_location_scale' of github.com:TuringLang/AdvancedVI.jl into proximal_entropy_location_scale
2 parents ce33a00 + d70a720 commit 3c5cb5a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/optimization/proximal_location_scale_entropy.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,7 @@ function apply(
5353
stepsize = stepsize_from_optimizer_state(leaf.rule, leaf.state)
5454
diag_idx = diagind(q.scale)
5555
scale_diag = q.scale[diag_idx]
56-
@. q.scale[diag_idx] =
57-
scale_diag + (sqrt(scale_diag^2 + 4 * stepsize) - scale_diag) / 2
56+
@. q.scale[diag_idx] = scale_diag + (sqrt(scale_diag^2 + 4 * stepsize) - scale_diag) / 2
5857

5958
params, _ = Optimisers.destructure(q)
6059

0 commit comments

Comments
 (0)