Correct shape for default grid_sizes

Summary: Small fix for omitting this argument.

Reviewed By: nikhilaravi

Differential Revision: D29548610

fbshipit-source-id: f25032fab3faa2f09006f5fcf8628138555f2f20
This commit is contained in:
Jeremy Reizenstein 2021-08-17 05:57:43 -07:00 committed by Facebook GitHub Bot
parent 46cf1970ac
commit b481cfbd01

View File

@ -181,8 +181,11 @@ def add_points_features_to_volume_densities_features(
# init the volumetric grid sizes if uninitialized
if grid_sizes is None:
grid_sizes = torch.LongTensor(list(volume_densities.shape[2:])).to(
volume_densities
# grid sizes shape (minibatch, 3)
grid_sizes = (
torch.LongTensor(list(volume_densities.shape[2:]))
.to(volume_densities)
.expand(volume_densities.shape[0], 3)
)
# flatten densities and features