423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-11-04 18:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix in reduction feature aggr.

Browse Source 
Summary: Bugfix

Reviewed By: bottler

Differential Revision: D40146840

fbshipit-source-id: a8415c361ed3cd939999b87311aff4d3bb1bcfe1
This commit is contained in:
David Novotny 2022-10-07 01:45:24 -07:00 committed by Facebook GitHub Bot
parent 4c8338b00f
commit 73ba66e4ab
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/implicitron/models/view_pooler/feature_aggregator.py
View File

@ -229,7 +229,7 @@ class ReductionFeatureAggregator(torch.nn.Module, FeatureAggregatorBase):
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,