metadata
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.safetensorsproj_head_text_infonce.safetensorsproj_head_text_mse.safetensorsconfig.json
Load through neurovlm.models.load_model("proj_head_image_infonce"),
load_model("proj_head_text_infonce"), or load_model("proj_head_text_mse").