Checkpoints
This directory holds all pretrained and finetuned model weights for DeepSparse.
Note: This directory is excluded from the git repository. Checkpoints are hosted on HuggingFace: HajihajihaJimmy/DeepSparse
Naming Convention
logs/
pretrain/ep_700.pth Pretrained foundation model (AtlasAbdomenMini, ep 700)
{dataset}+{n}v+s2/ep_400.pth Finetuned model — dataset, n input views, Stage 2, epoch 400
Checkpoint Map
| Folder | Anatomy | Views | Original training run |
|---|---|---|---|
pretrain/ |
— (pretraining) | 6–24 | atlas_7k+mv+rv |
abdomen+6v+s2/ |
Abdomen (PANORAMA) | 6 | atlas_7k+mv+rv+s1+ep7+abdomen+6v+s2 |
abdomen+8v+s2/ |
Abdomen (PANORAMA) | 8 | atlas_7k+mv+rv+s1+ep7+abdomen+8v+s2 |
abdomen+10v+s2/ |
Abdomen (PANORAMA) | 10 | atlas_7k+mv+rv+s1+ep7+abdomen+10v+s2 |
pelvis+6v+s2/ |
Pelvis (PENGWIN) | 6 | atlas_7k+mv+rv+s1+ep7+pelvis+6v+s2 |
pelvis+8v+s2/ |
Pelvis (PENGWIN) | 8 | atlas_7k+mv+rv+s1+ep7+pelvis+8v+s2 |
pelvis+10v+s2/ |
Pelvis (PENGWIN) | 10 | atlas_7k+mv+rv+s1+ep7+pelvis+10v+s2 |
tooth+6v+s2/ |
Tooth (ToothFairy) | 6 | atlas_7k+mv+rv+s1+ep7+tooth+6v+s2 |
tooth+8v+s2/ |
Tooth (ToothFairy) | 8 | atlas_7k+mv+rv+s1+ep7+tooth+8v+s2 |
tooth+10v+s2/ |
Tooth (ToothFairy) | 10 | atlas_7k+mv+rv+s1+ep7+tooth+10v+s2 |
luna+6v+s2/ |
Lung (LUNA16) | 6 | atlas_7k+mv+rv+ft_s1+ep7+s2 |
luna+8v+s2/ |
Lung (LUNA16) | 8 | atlas_7k+mv+rv+s1+ep7+8v+s2 |
luna+10v+s2/ |
Lung (LUNA16) | 10 | atlas_7k+mv+rv+s1+ep7+10v+s2 |
Usage
All evaluation scripts expect checkpoints at logs/{name}/ep_{epoch}.pth. The default epoch for inference is 400 (set via --epoch 400 in evaluate_s2.sh).