fix pointpillars flops calculation error (#1076)

open-mmlab · Dec 1, 2021 · bff52f4 · bff52f4
1 parent 3fd5ea2
commit bff52f4
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()