Skip to content

Commit a528305

Browse files
committed
access parameters correctly
1 parent 6f8679d commit a528305

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stochman/geodesic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def geodesic_minimizing_energy(curve, manifold, optimizer=torch.optim.Adam, max_
7777
# Initialize optimizer and set up closure
7878

7979
alpha = torch.linspace(0, 1, eval_grid, dtype=curve.begin.dtype, device=curve.device)
80-
opt = optimizer([curve.parameters], lr=1e-1)
80+
opt = optimizer(curve.parameters(), lr=1e-1)
8181

8282
def closure():
8383
opt.zero_grad()

0 commit comments

Comments
 (0)