IBL BWB
Collection
IBL BrainWideBench • 6 items • Updated
POYO+ · IBL BrainWideBench
Spike-token Perceiver, pretrained on 423 IBL Neuropixels sessions across all eight behavior tasks at once. Backbone for behavior decoding (TS1) and, after per-recording calibration, brain-region decoding (TS3). 12.4M parameters in the backbone, plus a 145.4M unit embedding table.
hf download nerdslab/ibl-bwb-poyo_plus poyo_plus.pt --local-dir ckpt
python src/ts1/train.py trainer=poyo_plus_gradual_unfreezing ckpt.load_from=ckpt/poyo_plus.pt \
recording_id=<EID> task=choice data_root=<PATH>
Loading a checkpoint with torch.load needs weights_only=False, since it carries its
run config alongside the weights.
Paper · Benchmark · Docs · Leaderboard · Training run
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support