From d879c03f968307288bbbea11f10bec00127ddc44 Mon Sep 17 00:00:00 2001 From: jacobpennington Date: Thu, 9 May 2024 14:57:59 -0700 Subject: [PATCH] Reverted special case for batch 0 in template_matching.py --- kilosort/template_matching.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/kilosort/template_matching.py b/kilosort/template_matching.py index 01ff2bc4..4d984a1e 100644 --- a/kilosort/template_matching.py +++ b/kilosort/template_matching.py @@ -45,8 +45,7 @@ def extract(ops, bfile, U, device=torch.device('cuda'), progress_bar=None): stt = stt.double() #st[k:k+nsp,0] = ((stt[:,0]-nt)/ops['fs'] + ibatch * (ops['batch_size']/ops['fs'])).cpu().numpy() - left_pad = 0 if ibatch == 0 else nt - st[k:k+nsp,0] = ((stt[:,0]-left_pad) + ibatch * (ops['batch_size'])).cpu().numpy() - nt//2 + ops['nt0min'] + st[k:k+nsp,0] = ((stt[:,0]-nt) + ibatch * (ops['batch_size'])).cpu().numpy() - nt//2 + ops['nt0min'] st[k:k+nsp,1] = stt[:,1].cpu().numpy() st[k:k+nsp,2] = amps[:,0].cpu().numpy()