File size: 504 Bytes
a0fcf71 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | ---
library_name: neurovlm
tags:
- neuroimaging
- contrastive-learning
- projection-heads
---
# ProjectionHeads
Projection heads used by NeuroVLM to align brain-map autoencoder latents and SPECTER text latents.
## Files
- `proj_head_image_infonce.safetensors`
- `proj_head_text_infonce.safetensors`
- `proj_head_text_mse.safetensors`
- `config.json`
Load through `neurovlm.models.load_model("proj_head_image_infonce")`,
`load_model("proj_head_text_infonce")`, or `load_model("proj_head_text_mse")`.
|