Skip to content

GitLab

"git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "4d535e885bf190478fb9293384351b615ae41b19"
Unverified Commit 87ae3d6f authored by loveSnowBest's avatar loveSnowBest Committed by GitHub
Browse files

[Enhancement] Enhance in vfe_layers creation (#389)

parent b820b0be
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/voxel_encoders/voxel_encoder.py
View file @ 87ae3d6f
...@@ -163,7 +163,7 @@ class DynamicVFE(nn.Module): ...@@ -163,7 +163,7 @@ class DynamicVFE(nn.Module):
nn.Sequential( nn.Sequential(
nn.Linear(in_filters, out_filters, bias=False), norm_layer, nn.Linear(in_filters, out_filters, bias=False), norm_layer,
nn.ReLU(inplace=True))) nn.ReLU(inplace=True)))
self.vfe_layers = nn.ModuleList(vfe_layers) self.vfe_layers = nn.ModuleList(vfe_layers)
self.num_vfe = len(vfe_layers) self.num_vfe = len(vfe_layers)
self.vfe_scatter = DynamicScatter(voxel_size, point_cloud_range, self.vfe_scatter = DynamicScatter(voxel_size, point_cloud_range,
(mode != 'max')) (mode != 'max'))
......
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