Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

segmentation models - make num_classes and softmax non must #2476

Open
yonigottesman opened this issue Aug 16, 2024 · 0 comments
Open

segmentation models - make num_classes and softmax non must #2476

yonigottesman opened this issue Aug 16, 2024 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@yonigottesman
Copy link

Currently all the segmentation models deep_lab_v3, segformer etc.. create the final layer with num_classes and always does softmax. Maybe we should have an activation parameter and if its None dont do any activation.
Some classification models require sigmoid and sometimes I want the unet for generative tasks without and activation

@mehtamansi29 mehtamansi29 added the enhancement New feature or request label Sep 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

3 participants