| # 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](https://huggingface.co/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`). |
|
|
