IBL BWB
Collection
IBL BrainWideBench • 6 items • Updated
NuCLR · IBL BrainWideBench
Contrastive pretraining on single-unit features, over 57,487 IBL Neuropixels units. Five seeds, each with the unit embeddings it produces, for brain-region decoding (TS3). 10.5M parameters per checkpoint.
hf download nerdslab/ibl-bwb-nuclr nuclr_embeddings_42.pt --local-dir embs
python src/ts3/eval.py probe=mlp emb_path=embs/nuclr_embeddings_42.pt
Seeds 42 to 46, as nuclr_<seed>.pt and nuclr_embeddings_<seed>.pt.
Loading a checkpoint with torch.load needs weights_only=False, since it carries its
run config alongside the weights.
Paper · Benchmark · Docs · Leaderboard · Training runs
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support