Fix bug: change the parameter name in PointNet2ClassificationSSG.__init__ from 'args' to 'hparams' so that users can use LightningModule.load_from_checkpoint load the checkpoints (#103)

Co-authored-by: Chao Huang <thss_huangc@163.com>

Author: matinjugou

pointnet2/models/pointnet2_ssg_cls.py

@@ -53,10 +53,10 @@ def load_state_dict(self, state):
 
 
 class PointNet2ClassificationSSG(pl.LightningModule):
-    def __init__(self, args):
+    def __init__(self, hparams):
         super().__init__()
 
-        self.hparams = args
+        self.hparams = hparams
 
         self._build_model()