Remove compile line in camera_utils (#2584)

temporarily remove torch.compile in camera_utils
This commit is contained in:
Gina Wu authored and GitHub committed 2023-11-05 10:21:16 -08:00
1 parent 85677d310f
commit 1f6fb8337c
1 file changed
+1 -2
+1 -2
View File
@@ -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,