fix pointpillars flops calculation error (#1076)

bff52f4b · ChaimZhu · GitHub · 3fd5ea26 · bff52f4b
Unverified Commit bff52f4b authored Dec 01, 2021 by ChaimZhu Committed by GitHub Dec 01, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet3d/models/voxel_encoders/pillar_encoder.py mmdet3d/models/voxel_encoders/pillar_encoder.py +1 -1

No files found.
--- a/mmdet3d/models/voxel_encoders/pillar_encoder.py
+++ b/mmdet3d/models/voxel_encoders/pillar_encoder.py
@@ -148,7 +148,7 @@ class PillarFeatureNet(nn.Module):
        for pfn in self.pfn_layers:
            features = pfn(features, num_points)

-        return features.squeeze()
+        return features.squeeze(1)


 @VOXEL_ENCODERS.register_module()