Thesis/pytorch-dnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Revert number of cores in search

Browse Source
This commit is contained in:
ixaxaar 2017-12-15 18:31:23 +05:30
parent e686d240ee
commit 95316a5b3b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dnc/flann_index.py
View File

@ -37,7 +37,7 @@ class FLANNIndex(object):
elif isinstance(query, T.Tensor):
query = query.cpu().numpy()
l, d = self.index.nn_index(query, num_neighbors=self.K if k is None else k, cores=4)
l, d = self.index.nn_index(query, num_neighbors=self.K if k is None else k)
distances = T.from_numpy(d).float()
labels = T.from_numpy(l).long()