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)

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Space using bilalahmad176176/BrainAge-Training-Code 1