tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

Fix torch.compile for torch 2.1 (#2577)

Browse Source
This commit is contained in:
Jonáš Kulhánek 2023-11-02 09:28:11 +01:00 committed by GitHub
parent 55616e1299
commit 85677d310f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/cameras/camera_utils.py
View File

@ -407,7 +407,7 @@ def _compute_residual_and_jacobian(
return fx, fy, fx_x, fx_y, fy_x, fy_y
@torch_compile(dynamic=True, mode="reduce-overhead", backend="eager")
@torch_compile(dynamic=True, mode="reduce-overhead")
def radial_and_tangential_undistort(
coords: torch.Tensor,
distortion_params: torch.Tensor,