From 4d2c022fe70a8604d65d24d72083337712f90fb2 Mon Sep 17 00:00:00 2001
From: ltriess <mail@triess.eu>
Date: Tue, 1 Jun 2021 17:19:36 +0200
Subject: [PATCH] Remove hard coded feature norm

---
 pointnet2/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pointnet2/model.py b/pointnet2/model.py
index a67e007..a63dba3 100644
--- 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
         ]