[Fix] Fix PointPillars FLOPs calculation error (#1075)

* fix pointpillars flops calculation error * fix

[Fix] Fix PointPillars FLOPs calculation error (#1075)
* fix pointpillars flops calculation error * fix
78a26540 · ChaimZhu · GitHub · 32747a54 · 78a26540
Unverified Commit 78a26540 authored Dec 01, 2021 by ChaimZhu Committed by GitHub Dec 01, 2021
Hide 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()