| 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")`. | |