-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[C++ API parity] Soft Margin loss #27660
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
Conversation
yf225
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks so much for the fantastic work @CarMiranda! I left some comments.
yf225
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@CarMiranda Thanks a lot for the awesome work! I will merge it after CI passes :D
|
@pytorchbot rebase this please |
|
TODO: wait for #27422 to land first, and then use |
|
@pytorchbot rebase this please |
facebook-github-bot
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@yf225 is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Summary: In accordance with pytorch#25883, I added the `SoftMarginLoss` module and `soft_margin_loss` functional. Pull Request resolved: pytorch#27660 Differential Revision: D17958325 Pulled By: yf225 fbshipit-source-id: c14422765e6e1fdabf6c9687080e6d5ff490d300
In accordance with #25883, I added the
SoftMarginLossmodule andsoft_margin_lossfunctional.