Merge pull request #2593 from alejoe91/ks4-batch-size-bug-fixes

Add batch size to KS4 wrapper
SpikeInterface · Mar 18, 2024 · f9ecdbf · f9ecdbf
2 parents 6cbe4c5 + af269ff
commit f9ecdbf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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.",