SpikeCascade / config.json
zeechimp's picture
Create config.json
d8ebacc verified
Raw History Blame Contribute Delete
435 Bytes
{
"architectures": ["SpikeCascadeForSorting"],
"model_type": "spike_cascade",
"sample_rate": 30000,
"n_channels": 384,
"spike_band_hz": [300, 6000],
"threshold_sigma": 4.5,
"template_bank_size": 256,
"template_len_samples": 61,
"template_confidence_threshold": 0.85,
"stage3_kmeans_k": 32,
"stage3_max_iter": 50,
"hidden_dim": 64,
"cnn_channels": [16, 32, 64],
"window_ms": 2.0,
"torch_dtype": "float32"
}