Skip to content

GitLab

Unverified Commit bff52f4b authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

fix pointpillars flops calculation error (#1076)

parent 3fd5ea26
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/voxel_encoders/pillar_encoder.py
View file @ bff52f4b
...@@ -148,7 +148,7 @@ class PillarFeatureNet(nn.Module): ...@@ -148,7 +148,7 @@ class PillarFeatureNet(nn.Module):
for pfn in self.pfn_layers: for pfn in self.pfn_layers:
features = pfn(features, num_points) features = pfn(features, num_points)
return features.squeeze() return features.squeeze(1)
@VOXEL_ENCODERS.register_module() @VOXEL_ENCODERS.register_module()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment