Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.Sign up
Wasserstein loss function #7406
What changes were proposed in this pull request?
Added the Wasserstein loss function.
How was this patch tested?
None. Should I add tests? I didn't see tests for any other losses except for crossentropy.
Note that the gradient doesn't match SameDiff's b/c of #7405.
AlexDBlack left a comment
Tests are in and the gradients pass.
The tests fail the serialization test because of an extra space at the end of the expected serialized string here:
OK, so the problem here is the equals method in DifferentialFunction.