diff --git a/dgcnn/dgcnn_train_pl.py b/dgcnn/dgcnn_train_pl.py index 3da7d3c04fb90b785dd40551e9fb93b1e64aae86..6a4bae7b55343cafec5613ed3e37a9e5b741e678 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'],