IBL BWB
Collection
IBL BrainWideBench • 6 items • Updated
POSSM · IBL BrainWideBench
Perceiver encoder over a recurrent backbone, 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). 11.2M parameters in the backbone, plus a 145.4M unit embedding table.
hf download nerdslab/ibl-bwb-possm possm.pt --local-dir ckpt
python src/ts1/train.py trainer=possm_gradual_unfreezing ckpt.load_from=ckpt/possm.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