diff --git a/models/segmentation_template/configs/train.yaml b/models/segmentation_template/configs/train.yaml index 943a06b2..3b28e700 100644 --- a/models/segmentation_template/configs/train.yaml +++ b/models/segmentation_template/configs/train.yaml @@ -144,7 +144,7 @@ inferer: # transform to apply to data from network to be suitable for validation postprocessing: _target_: Compose - transforms: + transforms: - _target_: Activationsd keys: '@pred' softmax: true