pointnet2 working - TODO: Eval!

utils/project_config.py

@@ -1,6 +1,6 @@
 from argparse import Namespace
 
-from utils.config import Config
+from ml_lib.utils.config import Config
 
 
 class GlobalVar(Namespace):
@@ -18,13 +18,16 @@ class GlobalVar(Namespace):
     DPI = 50
 
     # DATAOPTIONS
-    train='train',
-    vali='vali',
-    test='test'
+    train ='train',
+    vali ='vali',
+    test ='test'
+
+
+from models import *
 
 
 class ThisConfig(Config):
 
     @property
     def _model_map(self):
-        return dict()
+        return dict(PN2=PointNet2)