Downloads · 30 days
0
ibl-bwb-mtm
nerdslab/ibl-bwb-mtm
ibl-bwb-mtm is a machine learning model from nerdslab. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as mit.
Region-aware masked autoencoder, mixing neuron, causal, inter-region and intra-region masking, pretrained on 423 IBL Neuropixels sessions. Backbone for behavior decoding (TS1), activity prediction (TS2) and, after per…
Access
Public
Updated Sep 21, 2026
Repo size
1.7 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt839 MB · 100%
At a glance
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Jul 2, 2026
- Updated
- Sep 21, 2026
- SHA
- dfd7beac
Model card
From the Hugging Face model README
MtM · IBL BrainWideBench
Region-aware masked autoencoder, mixing neuron, causal, inter-region and intra-region 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.8M parameters in the encoder, plus 59.0M of per-session stitchers.
hf download nerdslab/ibl-bwb-mtm mtm.pt --local-dir ckpt
python src/ts1/train.py trainer=mtm_finetune ckpt.load_from=ckpt/mtm.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
- Library
- pytorch
- License
- mit
- Created
- Jul 2, 2026
- Updated
- Sep 21, 2026