Bugfix in reduction feature aggr.

Summary: Bugfix Reviewed By: bottler Differential Revision: D40146840 fbshipit-source-id: a8415c361ed3cd939999b87311aff4d3bb1bcfe1
facebookresearch · Oct 7, 2022 · 73ba66e · 73ba66e
1 parent 4c8338b
commit 73ba66e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py b/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py
@@ -229,7 +229,7 @@ def forward(
             masks_sampled.device,
             self.exclude_target_view,
         )
-        aggr_weigths = masks_sampled * sampling_mask
+        aggr_weigths = masks_sampled[..., 0] * sampling_mask[..., None]
         feats_aggregated = {
             k: _avgmaxstd_reduction_function(
                 f,