Skip to content

Commit

Permalink
Remove compile line in camera_utils (#2584)
Browse files Browse the repository at this point in the history
temporarily remove torch.compile in camera_utils
  • Loading branch information
ginazhouhuiwu authored Nov 5, 2023
1 parent 85677d3 commit 1f6fb83
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions nerfstudio/cameras/camera_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@
from numpy.typing import NDArray
from torch import Tensor

from nerfstudio.utils.misc import torch_compile

_EPS = np.finfo(float).eps * 4.0

Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 1f6fb83

Please sign in to comment.