Update deprecated PyTorch functions in fbcode/vision

Reviewed By: bottler Differential Revision: D58762015 fbshipit-source-id: a0d05fe63a88d33e3f7783b5a7b2a476dd3a7449
facebookresearch · Jun 20, 2024 · 00c36ec · 00c36ec
1 parent b0462d8
commit 00c36ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch3d/implicitron/dataset/frame_data.py b/pytorch3d/implicitron/dataset/frame_data.py
@@ -434,7 +434,7 @@ def collate(cls, batch):
             # TODO: don't store K; enforce working in NDC space
             return join_cameras_as_batch(batch)
         else:
-            return torch.utils.data._utils.collate.default_collate(batch)
+            return torch.utils.data.dataloader.default_collate(batch)
 
 
 FrameDataSubtype = TypeVar("FrameDataSubtype", bound=FrameData)