diff --git a/neurips23/sparse/nle/interface.py b/neurips23/sparse/nle/interface.py index 1a7a6181..42e2c33a 100644 --- 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,