Reverted special case for batch 0 in template_matching.py

MouseLand · May 9, 2024 · d879c03 · d879c03
1 parent ed9abbe
commit d879c03
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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()