nndet_run_training script#

Run nndet_train command to start nnDetection training for the specified fold.

usage: nndet_run_training [-h] --config-file CONFIG_FILE --train-config
                          TRAIN_CONFIG [--run-fold RUN_FOLD]
                          [--n-folds N_FOLDS] [--model MODEL]
                          [--resume-training RESUME_TRAINING]
                          [--n-workers N_WORKERS] [-v | -q]

Named Arguments#

--config-file

File path for the configuration dictionary, used to retrieve experiments variables (Task_ID)

--train-config

File path for the nnDetection Training YAML configuration dictionary, used to configure the nnDetection Experiment.

--run-fold

Index indicating which fold to run. Default: 0

Default: '0'

--n-folds

Number of Folds for final consolidation. Default: 5

Default: '5'

--model

nnDetection Model used for sweeping and consolidation. Default: RetinaUNetV001_D3V001_3d

Default: 'RetinaUNetV001_D3V001_3d'

--resume-training

Flag to indicate training resume after stopping it. Default no.

Default: no

--n-workers

Number of parallel processes used when pre-processing and unpacking the image data (Default: N_THREADS)

-v, --verbose

Enable verbose output in terminal. Add multiple times to increase verbosity.

-q, --silent

Suppress most log outputs in terminal.

Example call:

nndet_run_training --config-file ../CONFIG_FILE.json --run-fold 0
nndet_run_training --config-file ../CONFIG_FILE.json --run-fold 0 --resume-training y