softmax in UNET

[amida47]

is the UNET in monai.networks.nets.UNet does perform a softmax at end, or should I manually apply it to the predictions??

[KumoLiu]

Hi @amida47, please manually add it using Activations transform.
Please refer to:
https://github.com/Project-MONAI/tutorials/blob/main/3d_segmentation/brats_segmentation_3d.ipynb

Thanks.

[amida47]

I would like to suggest that a hint to the fact that the model doesn't do the softmax on it's own to the API reference, I was completely oblivious to to this and I went troubleshooting something else.