README.md

metadata

title: BrainAge Training
emoji: 🧠
colorFrom: green
colorTo: blue
sdk: docker
app_port: 7860
pinned: false
license: cc-by-nc-4.0

BrainAge Training Pipeline

GPU-powered training Space for the BrainAge brain-age prediction model.

Workflow:

1. Downloads preprocessed .pt cache from BrainAge-Golden-Preprocessed
2. Generates stratified train/val/test split
3. Phase A: Lifespan pretraining (all ages, 60 epochs)
4. Phase B: Pediatric fine-tuning (0–25y, 30 epochs)
5. Pushes checkpoints and logs to BrainAge-SFCN

Required secrets: HF_TOKEN (write access to model repo)