Commit 00c36ec0 authored by Sergii Dymchenko's avatar Sergii Dymchenko Committed by Facebook GitHub Bot
Browse files

Update deprecated PyTorch functions in fbcode/vision

Reviewed By: bottler

Differential Revision: D58762015

fbshipit-source-id: a0d05fe63a88d33e3f7783b5a7b2a476dd3a7449
parent b0462d80
...@@ -434,7 +434,7 @@ class FrameData(Mapping[str, Any]): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment