segmentation models - make num_classes
and softmax
non must
#2476
Labels
enhancement
New feature or request
num_classes
and softmax
non must
#2476
Currently all the segmentation models
deep_lab_v3
,segformer
etc.. create the final layer withnum_classes
and always doessoftmax
. Maybe we should have anactivation
parameter and if itsNone
dont do any activation.Some classification models require
sigmoid
and sometimes I want the unet for generative tasks without and activationThe text was updated successfully, but these errors were encountered: