Update interface.py

harsha-simhadri · Oct 19, 2023 · 1cb3c75 · 1cb3c75
1 parent edbb116
commit 1cb3c75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neurips23/sparse/nle/interface.py b/neurips23/sparse/nle/interface.py
@@ -133,7 +133,7 @@ def __init__(self,
     self.stops = stops
     self.scale = 100
     self.index_encoding = PISA_INDEX_DEFAULTS['index_encoding']
-    self.n_cores = cpu_count()
+    self.n_cores = 8
 
   def generate_forward(self):
     _pisathon.generate_fwd(self.path+"/inv")
@@ -256,7 +256,7 @@ def retrieve(self, index2, queries, k, k1, k2,
         stemmer='',
         queries=inp,
         k=k,
-        threads=self.n_cores,
+        threads=8,
         stop_fname='',
         query_weighted=1,
         pretokenised=True,