metrics in dgcnn implemented

dgcnn/dgcnn_train_pl.py

@@ -123,7 +123,7 @@ class DGCNNLightning(pl.LightningModule):
 
 # get train data 
 shapenet_data_train = ShapenetDataDgcnn(
-      root='/home/nibio/mutable-outside-world/code/oracle_gpu_runs/data/shapenet', 
+      root=config['data']['path'], 
       npoints=config['data']['npoints'],
       return_cls_label=True,
       small_data=config['data']['small_data'],
@@ -135,7 +135,7 @@ shapenet_data_train = ShapenetDataDgcnn(
 
 # get val data
 shapenet_data_val = ShapenetDataDgcnn(
-        root='/home/nibio/mutable-outside-world/code/oracle_gpu_runs/data/shapenet',
+        root=config['data']['path'], 
         npoints=config['data']['npoints'],
         return_cls_label=True,
         small_data=config['data']['small_data'],
@@ -147,7 +147,7 @@ shapenet_data_val = ShapenetDataDgcnn(
 
 # get test data
 shapenet_data_test = ShapenetDataDgcnn(
-        root='/home/nibio/mutable-outside-world/code/oracle_gpu_runs/data/shapenet',
+        root=config['data']['path'], 
         npoints=config['data']['npoints'],
         return_cls_label=True,
         small_data=config['data']['small_data'],