Add model card, safetensors weights, config, build_args and fsaverage5 mesh

.gitattributes

@@ -33,3 +33,18 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_coolwarm.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_dorsal.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_left.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_mosaic.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_rgb_overlay.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/brain_right.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/timesteps/frame_0001.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/timesteps/frame_0002.png filter=lfs diff=lfs merge=lfs -text
+e2e_output/timesteps/frame_0004.png filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/lh.infl filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/lh.pial filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/lh.white filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/rh.infl filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/rh.pial filter=lfs diff=lfs merge=lfs -text
+fsaverage5/surf/rh.white filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +1,140 @@
----
-license: cc-by-nc-4.0
----
+---
+license: cc-by-nc-4.0
+language:
+  - en
+tags:
+  - neuroscience
+  - fmri
+  - brain-encoding
+  - multimodal
+  - rust
+  - safetensors
+base_model: facebook/tribev2
+---
+
+<div align="center">
+
+# TRIBE v2 — Rust Edition
+
+**A Foundation Model of Vision, Audition, and Language for In-Silico Neuroscience**
+
+[![License: CC BY-NC 4.0](https://img.shields.io/badge/License-CC%20BY--NC%204.0-lightgrey.svg)](https://creativecommons.org/licenses/by-nc/4.0/)
+[![Rust](https://img.shields.io/badge/inference-Rust-orange.svg)](https://www.rust-lang.org/)
+[![Base model](https://img.shields.io/badge/base%20model-facebook%2Ftribev2-blue.svg)](https://huggingface.co/facebook/tribev2)
+
+📄 [Paper](https://ai.meta.com/research/publications/a-foundation-model-of-vision-audition-and-language-for-in-silico-neuroscience/) ·
+🤗 [Original weights](https://huggingface.co/facebook/tribev2) ·
+🦀 [Rust implementation](https://github.com/eugenehp/tribev2-rs)
+
+</div>
+
+## Overview
+
+This directory contains the **same pretrained weights** as [`facebook/tribev2`](https://huggingface.co/facebook/tribev2), converted to the [safetensors](https://github.com/huggingface/safetensors) format for use with the pure-Rust inference engine **tribev2-rs**.
+
+No fine-tuning, quantisation, or architectural changes have been made.  
+The model is **bit-for-bit equivalent** to the original Python checkpoint — every layer has been independently verified for numerical parity.
+
+## Model description
+
+TRIBE v2 is a deep multimodal brain encoding model that predicts fMRI responses to naturalistic stimuli (video, audio, text).  
+It combines three state-of-the-art feature extractors:
+
+| Modality | Extractor | Dim |
+|----------|-----------|----:|
+| Text | LLaMA 3.2-3B | 3 072 |
+| Audio | Wav2Vec-BERT 2.0 | 1 024 |
+| Video | V-JEPA2 ViT-G | 1 408 |
+
+These multimodal representations are projected and fused by a **Transformer encoder** (8 layers, 1 152-d, ScaleNorm, Rotary PE) that outputs predicted BOLD responses on the **fsaverage5** cortical mesh (~20 484 vertices).
+
+Full architectural details are in the [paper](https://ai.meta.com/research/publications/a-foundation-model-of-vision-audition-and-language-for-in-silico-neuroscience/) and in the [`facebook/tribev2`](https://huggingface.co/facebook/tribev2) model card.
+
+## Files
+
+| File | Description |
+|------|-------------|
+| `model.safetensors` | Pretrained weights (safetensors, converted from the original PyTorch Lightning checkpoint) |
+| `config.yaml` | Model hyper-parameters (hidden dim, depth, heads, modalities, …) |
+| `build_args.json` | Feature-extractor build arguments used at training time |
+| `fsaverage5/` | FreeSurfer fsaverage5 cortical mesh files (`.pial`, `.inflated`, `.sulc`, `.curv`) for brain visualisation |
+
+## Rust usage
+
+```rust
+use std::collections::BTreeMap;
+use tribev2_rs::model::tribe::TribeV2;
+use tribev2_rs::tensor::Tensor;
+
+// Load model from this data directory
+let model = TribeV2::from_pretrained(
+    "data/config.yaml",
+    "data/model.safetensors",
+    Some("data/build_args.json"),
+).unwrap();
+
+// Build multi-modal feature tensors [1, dim, T]
+let mut features = BTreeMap::new();
+features.insert("text".to_string(),  Tensor::zeros(&[1, 6144, 100]));
+features.insert("audio".to_string(), Tensor::zeros(&[1, 2048, 100]));
+features.insert("video".to_string(), Tensor::zeros(&[1, 2816, 100]));
+
+// Forward pass → [1, 20484, 100]
+let output = model.forward(&features, None, true);
+println!("{:?}", output.shape()); // [1, 20484, 100]
+```
+
+See the [tribev2-rs README](../README.md) for the full CLI, feature flags, benchmarks, and brain-visualisation API.
+
+## Converting weights from the original checkpoint
+
+```bash
+# 1. Download the original checkpoint from HuggingFace
+cargo run --bin tribev2-download --features hf-download -- --repo facebook/tribev2
+
+# 2. Convert to safetensors (requires Python ≥ 3.9, torch, safetensors)
+python3 scripts/convert_checkpoint.py weights/best.ckpt data/model.safetensors
+# → data/model.safetensors + data/build_args.json
+```
+
+## Pretrained model parameters
+
+| Parameter | Value |
+|-----------|-------|
+| Hidden dim | 1 152 |
+| Encoder depth | 8 |
+| Attention heads | 8 |
+| FF multiplier | 4× |
+| Norm | ScaleNorm |
+| Position encoding | Rotary (dim = 72) |
+| Low-rank head | 2 048 |
+| Subjects (released) | 1 (average subject) |
+| Output surface | fsaverage5 (20 484 vertices) |
+| Output timesteps | 100 TRs |
+
+## Citation
+
+If you use these weights or the Rust inference engine, please cite the original paper:
+
+```bibtex
+@article{dAscoli2026TribeV2,
+  title={A foundation model of vision, audition, and language for in-silico neuroscience},
+  author={d'Ascoli, St{\'e}phane and Rapin, J{\'e}r{\'e}my and Benchetrit, Yohann and
+          Brookes, Teon and Begany, Katelyn and Raugel, Jos{\'e}phine and
+          Banville, Hubert and King, Jean-R{\'e}mi},
+  year={2026}
+}
+```
+
+## License
+
+The **model weights** (all files in this directory) are released under the
+[Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0)](https://creativecommons.org/licenses/by-nc/4.0/) license,
+identical to the original [`facebook/tribev2`](https://huggingface.co/facebook/tribev2) release.
+
+> You are free to share and adapt the weights for **non-commercial** purposes,  
+> provided you give appropriate credit and indicate if changes were made.  
+> **Commercial use is not permitted.**
+
+The Rust source code of **tribev2-rs** is separately licensed under Apache-2.0.  
+See [LICENSE](../LICENSE) in the repository root.

build_args.json

@@ -0,0 +1,18 @@
+{
+  "feature_dims": {
+    "text": [
+      2,
+      3072
+    ],
+    "audio": [
+      2,
+      1024
+    ],
+    "video": [
+      2,
+      1408
+    ]
+  },
+  "n_outputs": 20484,
+  "n_output_timesteps": 100
+}

config.yaml

@@ -0,0 +1,619 @@
+infra:
+    folder: /checkpoint/sdascoli/results/tribe_release/half_depth/brain_model_config.subject_layers.subject_dropout=0.10,data.study.names=(Algonauts2025Bold,Lahner2024Bold,Lebel2023Bold,Wen2017)-ecdd89b2
+    cluster: auto
+    logs: '{folder}/logs/{user}/%j'
+    job_name: brain_model_config.subject_layers.subject_dropout=0.10,data.study.names=(Algonauts2025Bold,Lahner2024Bold,Lebel2023Bold,Wen2017)-ecdd89b2
+    timeout_min: 4320
+    nodes: 1
+    tasks_per_node: 1
+    cpus_per_task: 20
+    gpus_per_node: 1
+    mem_gb: 128.0
+    max_pickle_size_gb: null
+    slurm_constraint: volta32gb
+    slurm_partition: learnfair
+    slurm_account: null
+    slurm_qos: null
+    slurm_use_srun: false
+    slurm_additional_parameters: null
+    conda_env: null
+    workdir:
+        copied:
+        - neuralset
+        - neuraltrain
+        - neuralhub
+        - tribe
+        folder: null
+        log_commit: false
+        includes:
+        - '*.py'
+        - '*.txt'
+        excludes: !!python/tuple
+        - __pycache__
+        - .git
+    permissions: 511
+    version: '1'
+    mode: retry
+    keep_in_ram: false
+data:
+    study:
+        infra: null
+        names:
+        - Algonauts2025Bold
+        - Lahner2024Bold
+        - Lebel2023Bold
+        - Wen2017
+        path: /large_experiments/brainai/shared/studies
+        transforms:
+            extractaudio:
+                infra: null
+                overwrite: false
+                name: ExtractAudioFromVideo
+            extractwords:
+                infra: null
+                language: english
+                overwrite: false
+                name: ExtractWordsFromAudio
+            addtext:
+                infra: null
+                name: AddText
+            addsentence:
+                infra: null
+                max_unmatched_ratio: 0.05
+                override_sentences: false
+                name: AddSentenceToWords
+            addcontext:
+                infra: null
+                sentence_only: false
+                max_context_len: 1024
+                split_field: ''
+                name: AddContextToWords
+            removemissing:
+                infra: null
+                event_types: Word
+                field: context
+                name: RemoveMissing
+            chunksounds:
+                infra: null
+                event_type_to_chunk: Audio
+                event_type_to_use: null
+                min_duration: 30.0
+                max_duration: 60.0
+                name: ChunkEvents
+            chunkvideos:
+                infra:
+                    folder: !!python/object/apply:pathlib.PosixPath
+                    - /
+                    - checkpoint
+                    - sdascoli
+                    - cache
+                    - tribe_release
+                    cache_type: ValidatedParquet
+                    mode: cached
+                    keep_in_ram: false
+                    backend: Cached
+                event_type_to_chunk: Video
+                event_type_to_use: null
+                min_duration: 30.0
+                max_duration: 60.0
+                name: ChunkEvents
+            query:
+                infra: null
+                query: null
+                name: QueryEvents
+            split:
+                infra: null
+                val_ratio: 0.1
+                name: SplitEvents
+        query: null
+        studies_to_include: null
+        infra_timelines:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: processpool
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 720
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: null
+            gpus_per_node: null
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: null
+            slurm_partition: null
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: final
+            keep_in_ram: true
+            max_jobs: 1024
+            min_samples_per_job: 4
+            forbid_single_item_computation: false
+            mode: cached
+        name: MultiStudyLoader
+    neuro:
+        event_types: Fmri
+        aggregation: single
+        allow_missing: true
+        frequency: 1.0
+        offset: 5.0
+        projection:
+            mesh: fsaverage5
+            radius: 3.0
+            interpolation: linear
+            kind: ball
+            n_samples: null
+            mask_img: null
+            depth: null
+            center_depth: 0.5
+            extract_fsaverage_from_mni: true
+            name: TribeSurfaceProjector
+        cleaning:
+            standardize: zscore_sample
+            detrend: true
+            high_pass: null
+            low_pass: null
+            filter: null
+            ensure_finite: true
+        padding: null
+        input_space: null
+        fwhm: null
+        infra:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: slurm
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 720
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: 10
+            gpus_per_node: 1
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: volta32gb
+            slurm_partition: scavenge
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: release
+            keep_in_ram: true
+            max_jobs: 256
+            min_samples_per_job: 1
+            forbid_single_item_computation: false
+            mode: cached
+        name: FmriExtractor
+    text_feature:
+        model_name: meta-llama/Llama-3.2-3B
+        device: cuda
+        layers:
+        - 0.5
+        - 0.75
+        - 1.0
+        cache_all_layers: false
+        cache_n_layers: 20
+        layer_aggregation: group_mean
+        token_aggregation: mean
+        event_types: Word
+        aggregation: sum
+        allow_missing: true
+        frequency: 2.0
+        infra:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: slurm
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 720
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: 10
+            gpus_per_node: 1
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: volta32gb
+            slurm_partition: scavenge
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: release
+            keep_in_ram: true
+            max_jobs: 256
+            min_samples_per_job: 32
+            forbid_single_item_computation: false
+            mode: cached
+        batch_size: 4
+        contextualized: true
+        pretrained: true
+        name: HuggingFaceText
+    image_feature:
+        event_types: Video
+        aggregation: sum
+        allow_missing: true
+        frequency: 2.0
+        image:
+            model_name: facebook/dinov2-large
+            device: cuda
+            layers: 0.6666666666666666
+            cache_all_layers: false
+            cache_n_layers: 20
+            layer_aggregation: mean
+            token_aggregation: mean
+            event_types: Image
+            aggregation: single
+            allow_missing: false
+            frequency: 0.0
+            batch_size: 2
+            imsize: null
+            pretrained: true
+            infra:
+                folder: null
+                cluster: null
+                logs: '{folder}/logs/{user}/%j'
+                job_name: null
+                timeout_min: 25
+                nodes: 1
+                tasks_per_node: 1
+                cpus_per_task: 8
+                gpus_per_node: 1
+                mem_gb: null
+                max_pickle_size_gb: null
+                slurm_constraint: null
+                slurm_partition: null
+                slurm_account: null
+                slurm_qos: null
+                slurm_use_srun: false
+                slurm_additional_parameters: null
+                conda_env: null
+                workdir: null
+                permissions: 511
+                version: v5
+                keep_in_ram: false
+                max_jobs: 128
+                min_samples_per_job: 4096
+                forbid_single_item_computation: false
+                mode: cached
+            name: HuggingFaceImage
+        use_audio: true
+        clip_duration: null
+        max_imsize: null
+        layer_type: ''
+        num_frames: null
+        infra:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: slurm
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 1440
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: 10
+            gpus_per_node: 1
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: volta32gb
+            slurm_partition: scavenge
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: release
+            keep_in_ram: true
+            max_jobs: 1024
+            min_samples_per_job: 1
+            forbid_single_item_computation: false
+            mode: cached
+        name: HuggingFaceVideo
+    audio_feature:
+        model_name: facebook/w2v-bert-2.0
+        device: cuda
+        layers:
+        - 0.5
+        - 0.75
+        - 1.0
+        cache_all_layers: false
+        cache_n_layers: 20
+        layer_aggregation: group_mean
+        token_aggregation: mean
+        event_types: Audio
+        aggregation: sum
+        allow_missing: true
+        frequency: 2.0
+        norm_audio: true
+        infra:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: slurm
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 720
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: 10
+            gpus_per_node: 1
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: volta32gb
+            slurm_partition: scavenge
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: release
+            keep_in_ram: true
+            max_jobs: 256
+            min_samples_per_job: 1
+            forbid_single_item_computation: false
+            mode: cached
+        normalized: true
+        layer_type: transformer
+        name: Wav2VecBert
+    video_feature:
+        event_types: Video
+        aggregation: sum
+        allow_missing: true
+        frequency: 2.0
+        image:
+            model_name: facebook/vjepa2-vitg-fpc64-256
+            device: cuda
+            layers:
+            - 0.5
+            - 0.75
+            - 1.0
+            cache_all_layers: false
+            cache_n_layers: 20
+            layer_aggregation: group_mean
+            token_aggregation: mean
+            event_types: Image
+            aggregation: single
+            allow_missing: false
+            frequency: 0.0
+            batch_size: 8
+            imsize: null
+            pretrained: true
+            infra:
+                folder: null
+                cluster: null
+                logs: '{folder}/logs/{user}/%j'
+                job_name: null
+                timeout_min: 25
+                nodes: 1
+                tasks_per_node: 1
+                cpus_per_task: 8
+                gpus_per_node: 1
+                mem_gb: null
+                max_pickle_size_gb: null
+                slurm_constraint: null
+                slurm_partition: null
+                slurm_account: null
+                slurm_qos: null
+                slurm_use_srun: false
+                slurm_additional_parameters: null
+                conda_env: null
+                workdir: null
+                permissions: 511
+                version: v5
+                keep_in_ram: false
+                max_jobs: 128
+                min_samples_per_job: 4096
+                forbid_single_item_computation: false
+                mode: cached
+            name: HuggingFaceImage
+        use_audio: true
+        clip_duration: 4.0
+        max_imsize: null
+        layer_type: ''
+        num_frames: null
+        infra:
+            folder: /checkpoint/sdascoli/cache/tribe_release
+            cluster: slurm
+            logs: '{folder}/logs/{user}/%j'
+            job_name: null
+            timeout_min: 1440
+            nodes: 1
+            tasks_per_node: 1
+            cpus_per_task: 10
+            gpus_per_node: 1
+            mem_gb: null
+            max_pickle_size_gb: null
+            slurm_constraint: volta32gb
+            slurm_partition: scavenge
+            slurm_account: null
+            slurm_qos: null
+            slurm_use_srun: false
+            slurm_additional_parameters: null
+            conda_env: null
+            workdir: null
+            permissions: 511
+            version: release
+            keep_in_ram: true
+            max_jobs: 1024
+            min_samples_per_job: 1
+            forbid_single_item_computation: false
+            mode: cached
+        name: HuggingFaceVideo
+    subject_id:
+        event_types: Event
+        aggregation: first
+        allow_missing: true
+        frequency: 0.0
+        event_field: subject
+        treat_missing_as_separate_class: false
+        return_one_hot: false
+        predefined_mapping:
+            Algonauts2025Bold/sub-01: 0
+            Algonauts2025Bold/sub-02: 1
+            Algonauts2025Bold/sub-03: 2
+            Algonauts2025Bold/sub-05: 3
+            Lahner2024Bold/1: 4
+            Lahner2024Bold/10: 5
+            Lahner2024Bold/2: 6
+            Lahner2024Bold/3: 7
+            Lahner2024Bold/4: 8
+            Lahner2024Bold/5: 9
+            Lahner2024Bold/6: 10
+            Lahner2024Bold/7: 11
+            Lahner2024Bold/8: 12
+            Lahner2024Bold/9: 13
+            Lebel2023Bold/UTS01: 14
+            Lebel2023Bold/UTS02: 15
+            Lebel2023Bold/UTS03: 16
+            Lebel2023Bold/UTS04: 17
+            Lebel2023Bold/UTS05: 18
+            Lebel2023Bold/UTS06: 19
+            Lebel2023Bold/UTS07: 20
+            Lebel2023Bold/UTS08: 21
+            Wen2017/subject1: 22
+            Wen2017/subject2: 23
+            Wen2017/subject3: 24
+        name: LabelEncoder
+    frequency: 2.0
+    features_to_use:
+    - text
+    - audio
+    - video
+    features_to_mask: []
+    n_layers_to_use: null
+    layers_to_use:
+    - 0.5
+    - 0.75
+    - 1.0
+    layer_aggregation: group_mean
+    duration_trs: 100
+    overlap_trs_train: 0
+    overlap_trs_val: 0
+    batch_size: 8
+    num_workers: 20
+    shuffle_train: true
+    shuffle_val: true
+    stride_drop_incomplete: false
+    split_segments_by_time: false
+seed: 33
+brain_model_config:
+    projector:
+        input_size: null
+        hidden_sizes: null
+        norm_layer: layer
+        activation_layer: gelu
+        bias: true
+        dropout: 0.0
+        name: Mlp
+    combiner: null
+    encoder:
+        heads: 8
+        depth: 8
+        cross_attend: false
+        causal: false
+        attn_flash: false
+        attn_dropout: 0.0
+        ff_mult: 4
+        ff_dropout: 0.0
+        use_scalenorm: true
+        use_rmsnorm: false
+        rel_pos_bias: false
+        alibi_pos_bias: false
+        rotary_pos_emb: true
+        rotary_xpos: false
+        residual_attn: false
+        scale_residual: true
+        layer_dropout: 0.0
+        name: TransformerEncoder
+    time_pos_embedding: true
+    subject_embedding: false
+    subject_layers:
+        n_subjects: 25
+        bias: true
+        init_id: false
+        average_subjects: false
+        mode: gather
+        subject_dropout: 0.1
+        name: SubjectLayers
+    hidden: 1152
+    max_seq_len: 1024
+    dropout: 0.0
+    extractor_aggregation: cat
+    layer_aggregation: cat
+    linear_baseline: false
+    modality_dropout: 0.3
+    temporal_dropout: 0.0
+    low_rank_head: 2048
+    temporal_smoothing: null
+    name: FmriEncoder
+loss:
+    kwargs:
+        reduction: none
+    name: MSELoss
+optim:
+    optimizer:
+        lr: 0.0001
+        kwargs:
+            weight_decay: 0.0
+        name: Adam
+    scheduler:
+        kwargs:
+            max_lr: 0.0001
+            pct_start: 0.1
+        name: OneCycleLR
+    interval: step
+    name: LightningOptimizer
+metrics:
+-   log_name: pearson
+    dim: 0
+    reduction: mean
+    torchmetrics_kwargs: null
+    name: OnlinePearsonCorr
+-   log_name: subj_pearson
+    metric_name: OnlinePearsonCorr
+    kwargs:
+        dim: 0
+    name: GroupedMetric
+-   log_name: retrieval_top1
+    topk: 1
+    name: TopkAcc
+monitor: val/pearson
+wandb_config:
+    name: brain_model_config.subject_layers.subject_dropout=0.10,data.study.names=(Algonauts2025Bold,Lahner2024Bold,Lebel2023Bold,Wen2017)-ecdd89b2
+    group: half_depth
+    entity: sdascoli
+    project: tribe_release
+    offline: false
+    host: https://fairwandb.org
+    id: null
+    dir: null
+    anonymous: null
+    log_model: false
+    experiment: null
+    prefix: ''
+    resume: allow
+accelerator: gpu
+n_epochs: 15
+max_steps: -1
+patience: null
+limit_train_batches: null
+accumulate_grad_batches: 1
+enable_progress_bar: true
+log_every_n_steps: 5
+fast_dev_run: false
+save_checkpoints: true
+checkpoint_filename: best
+resize_subject_layer: false
+freeze_backbone: false
+average_subjects: false
+checkpoint_path: null
+load_checkpoint: true
+test_only: false

e2e_output/predictions.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:822ef4161933ddf101970c19ebfabcb55b42393cbeb82847060baab38da95125
+size 8193600

e2e_output/predictions_segmented.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7aaebd4979b25ea967599e6f303991c6754aeb85c447942534289cc7d8e457f
+size 16387200

fsaverage5/surf/lh.infl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c521035f63cff7eb9b2be1778ec7db5accfa7247f4e0626ecf06f147eedf1192
+size 368720

fsaverage5/surf/lh.pial

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34cccc5b7f64aaef670beb90ae3423aa258c2d6d81275fc8372d9a34ed97fb83
+size 368720

fsaverage5/surf/lh.white

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46756fc30763b48ca69fc4b2bd0a2c7fe640a5f32f2d8920f63bd42c59f773f0
+size 368720

fsaverage5/surf/rh.infl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd829b1b09a1324a71b9a5a7ba4a9dfb2c67ca25c906e5453cdac7e1cd29bc79
+size 368720

fsaverage5/surf/rh.pial

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:174fc904af97a4a38619355e3211427a57ae336922e5b941a15f2a0d7daabf50
+size 368720

fsaverage5/surf/rh.white

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fafaff5f1cadc4342ac6b9c7b3cfc9884d208f57bab8815edf0c6097eab303e5
+size 368720

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c48e25476616b955eb4ba40ad65200e9688b29ec050f72bdd9492471bbf946a
+size 708832652