You need to sign in or sign up before continuing.
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