deploy pointnet++ finished

modules/pointnet++_encoder.py

@@ -7,6 +7,7 @@ for i in range(2):
     path = os.path.dirname(path)
 PROJECT_ROOT = path                 
 sys.path.append(PROJECT_ROOT)
+import PytorchBoot.stereotype as stereotype
 from modules.module_lib.pointnet2_modules import PointnetSAModuleMSG
 
 
@@ -66,8 +67,9 @@ def break_up_pc(pc):
     return xyz, features
 
 
+@stereotype.module("pointnet++_encoder")
 class PointNet2Encoder(nn.Module):
-    def encode_points(self, pts):
+    def encode_points(self, pts, require_per_point_feat=False):
         return self.forward(pts)
 
     def __init__(self, config:dict):