IBL BWB
Collection
IBL BrainWideBench • 6 items • Updated
NDT Stitch · IBL BrainWideBench
Masked-autoencoder Transformer with block masking, pretrained on 423 IBL Neuropixels sessions. Backbone for behavior decoding (TS1), activity prediction (TS2) and, after per-recording calibration, brain-region decoding (TS3). 10.3M parameters in the encoder, plus 32.7M of per-session stitchers.
hf download nerdslab/ibl-bwb-ndt_stitch ndt_stitch.pt --local-dir ckpt
python src/ts1/train.py trainer=ndt_stitch_finetune ckpt.load_from=ckpt/ndt_stitch.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