Skip to content

Commit 09d66e2

Browse files
committed
drop ncs jobs to 8, we see some memory creep with 16
1 parent a08d6f3 commit 09d66e2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ulc_mm_package/neural_nets/NCSModel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def __init__(
8787
self._temp_infer_queue = AsyncInferQueue(self.model)
8888

8989
# used for asyn
90-
self.asyn_infer_queue = AsyncInferQueue(self.model, jobs=16)
90+
self.asyn_infer_queue = AsyncInferQueue(self.model, jobs=8)
9191
self.asyn_infer_queue.set_callback(self._default_callback)
9292
self._asyn_results: List[
9393
Union[AsyncInferenceResult, List[AsyncInferenceResult]]

0 commit comments

Comments
 (0)