-
Notifications
You must be signed in to change notification settings - Fork 26.3k
C++ API parity: MarginRankingLoss #29000
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
[ghstack-poisoned]
[ghstack-poisoned]
Differential Revision: [D18271855](https://our.internmc.facebook.com/intern/diff/D18271855) [ghstack-poisoned]
Differential Revision: [D18271855](https://our.internmc.facebook.com/intern/diff/D18271855) [ghstack-poisoned]
Differential Revision: [D18271855](https://our.internmc.facebook.com/intern/diff/D18271855) [ghstack-poisoned]
Differential Revision: [D18271855](https://our.internmc.facebook.com/intern/diff/D18271855) [ghstack-poisoned]
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.
@pbelevich Thanks so much for the awesome work! I have one very minor comment
| typedef c10::variant<enumtype::kNone, enumtype::kMean, enumtype::kSum> reduction_t; | ||
|
|
||
| /// Has a default value of `0`. | ||
| TORCH_ARG(float, margin) = 0; |
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.
I believe we should use
| TORCH_ARG(float, margin) = 0; | |
| TORCH_ARG(double, margin) = 0; |
because float in Python is always 64-bit (there are a few other places in the C++ API where we use TORCH_ARG(float, ...), and we will need to change them as well :/ (Not part of this PR))
Differential Revision: [D18271855](https://our.internmc.facebook.com/intern/diff/D18271855) [ghstack-poisoned]
|
@pbelevich merged this pull request in 69f845c. |
Stack from ghstack:
Differential Revision: D18271855