Skip to content

GitLab

"docs/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "a4211a4dc3a83d9e58eb7ee2f015aa033159c267"
Unverified Commit 78a26540 authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

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

* fix pointpillars flops calculation error

* fix
parent 32747a54
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/voxel_encoders/pillar_encoder.py
View file @ 78a26540
...@@ -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