Skip to content

Commit

Permalink
Merge pull request #684 from umrlastig/fix-torch-solve
Browse files Browse the repository at this point in the history
Replace torch.solve by torch.linalg.solve (torch 1.8)
  • Loading branch information
nicolas-chaulet committed Oct 11, 2021
2 parents 84251ec + 7e0f909 commit 2495ec4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_points3d/utils/registration.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def fast_global_registration(xyz, xyz_target, mu_init=1, num_iter=20):
if i > 0 and i % 5 == 0:
mu /= 2.0
A, b = get_matrix_system(source, xyz_target, weight)
x, _ = torch.solve(A.T @ b, A.T.mm(A))
x = torch.linalg.solve(A.T.mm(A), A.T @ b)
T = get_trans(x.view(-1))
source = source.mm(T[:3, :3].T) + T[:3, 3]
T_res = T @ T_res
Expand Down

0 comments on commit 2495ec4

Please sign in to comment.