diff --git a/pytorch3d/implicitron/dataset/frame_data.py b/pytorch3d/implicitron/dataset/frame_data.py index e4321549..87ec9084 100644 --- 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)