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.

Sign up for GitHub

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

Jump to bottom

Expand criterion pool to support various types of losses #30

Open
TibbersHao opened this issue Sep 5, 2024 · 0 comments
Open

Expand criterion pool to support various types of losses #30

TibbersHao opened this issue Sep 5, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@TibbersHao
Copy link
Member

Currently the build_criterion function in train.py only supports CrossEntropy loss, this needs to be expanded to support additional loss options such as MSE, Binary Entropy etc. As for a temporary solution, frontend will only display CrossEntropy loss for the default option until this feature upgrade is done.
@TibbersHao TibbersHao added the enhancement New feature or request label Sep 5, 2024
@TibbersHao TibbersHao mentioned this issue Sep 5, 2024
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@TibbersHao