[Enhancement] Enhance in vfe_layers creation (#389)

87ae3d6f · loveSnowBest · GitHub · b820b0be · 87ae3d6f
Unverified Commit 87ae3d6f authored Mar 30, 2021 by loveSnowBest Committed by GitHub Mar 30, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet3d/models/voxel_encoders/voxel_encoder.py mmdet3d/models/voxel_encoders/voxel_encoder.py +1 -1

No files found.
--- a/mmdet3d/models/voxel_encoders/voxel_encoder.py
+++ b/mmdet3d/models/voxel_encoders/voxel_encoder.py
@@ -163,7 +163,7 @@ class DynamicVFE(nn.Module):
                nn.Sequential(
                    nn.Linear(in_filters, out_filters, bias=False), norm_layer,
                    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.vfe_scatter = DynamicScatter(voxel_size, point_cloud_range,
                                          (mode != 'max'))