remove profiling

nerfstudio-project · Jan 13, 2025 · 6270e6b · 6270e6b
1 parent 37f4ca4
commit 6270e6b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nerfstudio/data/utils/data_utils.py b/nerfstudio/data/utils/data_utils.py
@@ -22,7 +22,6 @@
 import torch
 from PIL import Image
 from PIL.Image import Image as PILImage
-from torch.profiler import record_function
 
 
 def pil_to_numpy(im: PILImage) -> np.ndarray:
@@ -35,8 +34,7 @@ def pil_to_numpy(im: PILImage) -> np.ndarray:
         numpy.ndarray representing the image data.
     """
     # Load in image completely (PIL defaults to lazy loading)
-    with record_function("im.load()"):
-        im.load()
+    im.load()
 
     # Unpack data
     e = Image._getencoder(im.mode, "raw", im.mode)