Commit 9db70400 authored by Pyre Bot Jr's avatar Pyre Bot Jr Committed by Facebook GitHub Bot
Browse files

suppress errors in `fbcode/vision` - batch 2

Differential Revision: D30222339

fbshipit-source-id: 97d498df72ef897b8dc2405764e3ffd432082e3c
parent 80411783
...@@ -123,7 +123,7 @@ class GridRaysampler(torch.nn.Module): ...@@ -123,7 +123,7 @@ class GridRaysampler(torch.nn.Module):
device = cameras.device device = cameras.device
# expand the (H, W, 2) grid batch_size-times to (B, H, W, 2) # expand the (H, W, 2) grid batch_size-times to (B, H, W, 2)
xy_grid = self._xy_grid.to(device)[None].expand( # pyre-ignore xy_grid = self._xy_grid.to(device)[None].expand(
batch_size, *self._xy_grid.shape batch_size, *self._xy_grid.shape
) )
......
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