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