Remove hard coded feature norm

ltriess · Jun 1, 2021 · 4d2c022 · 4d2c022
1 parent 7fcaf2a
commit 4d2c022
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pointnet2/model.py b/pointnet2/model.py
@@ -134,7 +134,7 @@ def __init__(
             raise ValueError(f"Received unknown feature norm `{feature_norm}`!")
 
         self.segmentation_layers = [
-            FeaturePropagationModule(mlp_units=units, feature_norm="batch")
+            FeaturePropagationModule(mlp_units=units, feature_norm=feature_norm)
             for units in fp_units
         ]