Skip to content

Commit fe893b0

Browse files
committed
slightly more standardized
1 parent 46f799f commit fe893b0

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

test/manifolds/manifolddiff.jl

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,21 @@ data = [exp(M, p, σ * rand(M; vector_at=p)) for i in 1:n];
2424

2525
# objective function
2626
f(M, p) = sum(1 / (2 * n) * distance.(Ref(M), Ref(p), data) .^ 2)
27-
f_(p) = f(M,p)
27+
# f_(p) = f(M,p)
2828

29+
# non-manual: intrinsic finite differences gradient
30+
function grad_f_FD(M,p)
31+
f_(p_) = f(M,p_)
32+
ManifoldDiff.gradient(M, f_, p, r_backend)
33+
end
2934
# manual gradient
3035
# grad_f(M, p) = sum(1 / n * grad_distance.(Ref(M), data, Ref(p)));
3136

32-
# non-manual: intrinsic finite differences gradient
33-
grad_f_FD(M,p) = ManifoldDiff.gradient(M, f_, p, r_backend)
3437

3538
# and solve
36-
m1 = gradient_descent(M, f, grad_f_FD, data[1])
39+
@time m1 = gradient_descent(M, f, grad_f_FD, data[1])
3740

41+
@info "Basic Manopt test" string(m1')
3842
@test isapprox(p, m1; atol=0.15)
3943

4044
##

0 commit comments

Comments
 (0)