From 1f6fb8337c05afb357d61815416266a591966c61 Mon Sep 17 00:00:00 2001 From: Gina Wu <42229107+ginazhouhuiwu@users.noreply.github.com> Date: Sun, 5 Nov 2023 10:21:16 -0800 Subject: [PATCH] Remove compile line in camera_utils (#2584) temporarily remove torch.compile in camera_utils --- nerfstudio/cameras/camera_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nerfstudio/cameras/camera_utils.py b/nerfstudio/cameras/camera_utils.py index 40419528a..ffb557589 100644 --- a/nerfstudio/cameras/camera_utils.py +++ b/nerfstudio/cameras/camera_utils.py @@ -25,7 +25,6 @@ from jaxtyping import Float from numpy.typing import NDArray from torch import Tensor -from nerfstudio.utils.misc import torch_compile _EPS = np.finfo(float).eps * 4.0 @@ -407,7 +406,7 @@ def _compute_residual_and_jacobian( return fx, fy, fx_x, fx_y, fy_x, fy_y -@torch_compile(dynamic=True, mode="reduce-overhead") +# @torch_compile(dynamic=True, mode="reduce-overhead", backend="eager") def radial_and_tangential_undistort( coords: torch.Tensor, distortion_params: torch.Tensor,