Update deprecated PyTorch functions in fbcode/vision

Reviewed By: bottler

Differential Revision: D58762015

fbshipit-source-id: a0d05fe63a88d33e3f7783b5a7b2a476dd3a7449
This commit is contained in:
Sergii Dymchenko 2024-06-20 14:06:28 -07:00 committed by Facebook GitHub Bot
parent b0462d8079
commit 00c36ec01c

View File

@ -434,7 +434,7 @@ class FrameData(Mapping[str, Any]):
# TODO: don't store K; enforce working in NDC space # TODO: don't store K; enforce working in NDC space
return join_cameras_as_batch(batch) return join_cameras_as_batch(batch)
else: else:
return torch.utils.data._utils.collate.default_collate(batch) return torch.utils.data.dataloader.default_collate(batch)
FrameDataSubtype = TypeVar("FrameDataSubtype", bound=FrameData) FrameDataSubtype = TypeVar("FrameDataSubtype", bound=FrameData)