Skip to content

Commit 8e510e1

Browse files
Update docstring in get_nnunet_trainer to include link for supported trainer classes
1 parent 88a28d2 commit 8e510e1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

monai/apps/nnunet/nnunet_bundle.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ def get_nnunet_trainer(
8282
fold : Union[int, str]
8383
The fold number or 'all' for cross-validation.
8484
trainer_class_name : str, optional
85-
The class name of the trainer to be used. Default is 'nnUNetTrainer'.
85+
The class name of the trainer to be used. Default is 'nnUNetTrainer'.
86+
For a complete list of supported trainers, see https://github.com/MIC-DKFZ/nnUNet/tree/master/nnunetv2/training/nnUNetTrainer/variants
8687
plans_identifier : str, optional
8788
Identifier for the plans to be used. Default is 'nnUNetPlans'.
8889
use_compressed_data : bool, optional

0 commit comments

Comments
 (0)