Skip to content

Conversation

@pbelevich
Copy link
Contributor

@pbelevich pbelevich commented Oct 31, 2019

Stack from ghstack:

Differential Revision: D18271855

Copy link
Contributor

@yf225 yf225 left a 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;
Copy link
Contributor

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

Suggested change
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))

@pbelevich pbelevich requested a review from yf225 November 4, 2019 22:21
@facebook-github-bot
Copy link
Contributor

@pbelevich merged this pull request in 69f845c.

@facebook-github-bot facebook-github-bot deleted the gh/pbelevich/52/head branch November 8, 2019 15:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants