From 73ba66e4abb14abd328d2e96f572337e04770cda Mon Sep 17 00:00:00 2001 From: David Novotny Date: Fri, 7 Oct 2022 01:45:24 -0700 Subject: [PATCH] Bugfix in reduction feature aggr. Summary: Bugfix Reviewed By: bottler Differential Revision: D40146840 fbshipit-source-id: a8415c361ed3cd939999b87311aff4d3bb1bcfe1 --- pytorch3d/implicitron/models/view_pooler/feature_aggregator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py b/pytorch3d/implicitron/models/view_pooler/feature_aggregator.py index 442466bf9..b51cd0c05 100644 --- 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,