From 2a6c22bb76cd950ed87b992f7247262f0fb65e1c Mon Sep 17 00:00:00 2001
From: Maciej Wielgosz <maciej.wielgosz@nibio.no>
Date: Tue, 28 Mar 2023 12:21:44 +0200
Subject: [PATCH] metrics in dgcnn implemented

---
 dgcnn/dgcnn_train_pl.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/dgcnn/dgcnn_train_pl.py b/dgcnn/dgcnn_train_pl.py
index 3da7d3c..6a4bae7 100644
--- a/dgcnn/dgcnn_train_pl.py
+++ b/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'],
-- 
GitLab