Skip to content

Commit

Permalink
add train instructions
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianIsensee committed Jun 3, 2024
1 parent 604d0fd commit 19ae215
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions nnunetv2/dataset_conversion/Dataset224_AbdomenAtlas1.0.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,20 @@


if __name__ == '__main__':
"""
How to train our submission to the JHU benchmark
1. Execute this script here to convert the dataset into nnU-Net format. Adapt the paths to your system!
2. Run planning and preprocessing: `nnUNetv2_plan_and_preprocess -d 224 -npfp 64 -np 64 -c 3d_fullres -pl
nnUNetPlannerResEncL_torchres`. Adapt the number of processes to your System (-np; -npfp)! Note that each process
will again spawn 4 threads for resampling. This custom planner replaces the nnU-Net default resampling scheme with
a torch-based implementation which is faster but less accurate. This is needed to satisfy the inference speed
constraints.
3. Run training with `nnUNetv2_train 224 3d_fullres all -p nnUNetResEncUNetLPlans_torchres`. 24GB VRAM required,
training will take ~28-30h.
"""


base = '/home/isensee/Downloads/AbdomenAtlas1.0Mini'
cases = subdirs(base, join=False, prefix='BDMAP')

Expand Down

0 comments on commit 19ae215

Please sign in to comment.