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:
0Default:
'0'- --n-folds
Number of Folds for final consolidation. Default:
5Default:
'5'- --model
nnDetection Model used for sweeping and consolidation. Default:
RetinaUNetV001_D3V001_3dDefault:
'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