From af269ff051486f5d7d141f269d51c5f86c9ccf79 Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Mon, 18 Mar 2024 10:05:56 +0100 Subject: [PATCH] Add batch size to KS4 wrapper --- src/spikeinterface/sorters/external/kilosort4.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/spikeinterface/sorters/external/kilosort4.py b/src/spikeinterface/sorters/external/kilosort4.py index 9f7ce929c9..99de91a795 100644 --- a/src/spikeinterface/sorters/external/kilosort4.py +++ b/src/spikeinterface/sorters/external/kilosort4.py @@ -18,6 +18,7 @@ class Kilosort4Sorter(BaseSorter): gpu_capability = "nvidia-optional" _default_params = { + "batch_size": 60000, "nblocks": 1, "Th_universal": 9, "Th_learned": 8, @@ -53,6 +54,7 @@ class Kilosort4Sorter(BaseSorter): } _params_description = { + "batch_size": "Number of samples per batch. Default value: 60000.", "nblocks": "Number of non-overlapping blocks for drift correction (additional nblocks-1 blocks are created in the overlaps). Default value: 1.", "Th_universal": "Spike detection threshold for universal templates. Th(1) in previous versions of Kilosort. Default value: 9.", "Th_learned": "Spike detection threshold for learned templates. Th(2) in previous versions of Kilosort. Default value: 8.",