Fix Now Working

Stelath · Sep 7, 2023 · 5b56c8a · 5b56c8a
1 parent e7a135a
commit 5b56c8a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -9,3 +9,4 @@ weights/*
 pretrained
 data/Kitti/downsampled
 data/ModelNet/*.pkl
+data/3DMatch/data/*
diff --git a/geotransformer/utils/pointcloud.py b/geotransformer/utils/pointcloud.py
@@ -15,7 +15,7 @@ def get_nearest_neighbor(
 ):
     r"""Compute the nearest neighbor for the query points in support points."""
     s_tree = cKDTree(s_points)
-    distances, indices = s_tree.query(q_points, k=1, n_jobs=-1)
+    distances, indices = s_tree.query(q_points, k=1)
     if return_index:
         return distances, indices
     else:

diff --git a/load_cuda.sh b/load_cuda.sh
@@ -0,0 +1 @@
+module load cuda/12.0.1-gcc
diff --git a/requirements.txt b/requirements.txt
@@ -5,7 +5,7 @@ tqdm
 coloredlogs
 easydict
 nibabel
-open3d==0.11.2
+open3d==0.14.1
 scikit-learn
 einops
 ipdb