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

* fix pointpillars flops calculation error * fix
open-mmlab · Dec 1, 2021 · 78a2654 · 78a2654
1 parent 32747a5
commit 78a2654
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet3d/models/voxel_encoders/pillar_encoder.py b/mmdet3d/models/voxel_encoders/pillar_encoder.py
@@ -148,7 +148,7 @@ def forward(self, features, num_points, coors):
         for pfn in self.pfn_layers:
             features = pfn(features, num_points)
 
-        return features.squeeze()
+        return features.squeeze(1)
 
 
 @VOXEL_ENCODERS.register_module()