diff --git a/src/spikeinterface/sorters/external/kilosort2_master.m b/src/spikeinterface/sorters/external/kilosort2_master.m index da7c5f5598..1e2587aac2 100644 --- a/src/spikeinterface/sorters/external/kilosort2_master.m +++ b/src/spikeinterface/sorters/external/kilosort2_master.m @@ -26,7 +26,7 @@ function kilosort2_master(fpath, kilosortPath) if skip_kilosort_preprocessing % hack to skip the internal preprocessing % this mimic the preprocessDataSub() function - fprintf("SKIP kilosort2.5 preprocessing\n"); + fprintf("Skipping kilosort2 preprocessing\n"); ops.nt0 = getOr(ops, {'nt0'}, 61); % number of time samples for the templates (has to be <=81 due to GPU shared memory) ops.nt0min = getOr(ops, 'nt0min', ceil(20 * ops.nt0/61)); % time sample where the negative peak should be aligned @@ -48,12 +48,14 @@ function kilosort2_master(fpath, kilosortPath) rez.yc = yc; rez.xcoords = xc; rez.ycoords = yc; - Nbatch = ceil(ops.sampsToRead / ops.NT); % number of data batches - NTbuff = ops.NT + 3*ops.ntbuff; % we need buffers on both sides for filtering + Nbatch = ceil(ops.sampsToRead /(NT-ops.ntbuff)); % number of data batches + ops.Nbatch = Nbatch; + NTbuff = NT + 4*ops.ntbuff; % we need buffers on both sides for filtering rez.Wrot = eye(ops.Nchan); % fake whitenning rez.temp.Nbatch = Nbatch; - % fproc is the same as the binary + % fproc (preprocessed output) is the same as the fbinary (unprocessed input) + % because we are skipping preprocessing. ops.fproc = ops.fbinary; rez.ops = ops; % memorize ops @@ -62,7 +64,7 @@ function kilosort2_master(fpath, kilosortPath) rez.ops.Nbatch = Nbatch; rez.ops.NTbuff = NTbuff; - tic; % tocs are coming + tic; % tocs are coming TODO else % preprocess data to create temp_wh.dat