Implementation of the loss from Supervised Contrastive Loss paper.

[Nimrais]

Before submitting (checklist)

• [x] Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• [x] Did you read the contribution guide?
• [x] Did you check the code style? catalyst-make-codestyle && catalyst-check-codestyle (pip install -U catalyst-codestyle).
• [x] Did you make sure to update the docs? We use Google format for all the methods and classes.
• [ ] Did you check the docs with make check-docs?
• [x] Did you write any new necessary tests?
• [x] Did you check that your code passes the unit tests pytest . ?
• [ ] Did you add your new functionality to the docs?
• [x] Did you update the CHANGELOG?
• [ ] Did you run colab minimal CI/CD with latest and minimal requirements?
• [ ] Did you check XLA integration with single and multiple processes?

Description

In the paper Supervised Contrastive Loss, the abstract contrastive criterion was proposed. This PR carefully implements this criterion.

Related Issue

Type of Change

• [x] Examples / docs / tutorials / contributors update
• [ ] Bug fix (non-breaking change which fixes an issue)
• [ ] Improvement (non-breaking change which improves an existing feature)
• [x] New feature (non-breaking change which adds functionality)
• [ ] Breaking change (fix or feature that would cause existing functionality to change)

PR review

Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

FAQ

Please review the FAQ before submitting an issue:

• [x] I have read the documentation and FAQ
• [x] I have reviewed the minimal examples section
• [x] I have checked the changelog for main framework updates
• [x] I have read the contribution guide
• [x] I have joined Catalyst slack (#__questions channel) for issue discussion

[pep8speaks]

Hello @Nimrais! Thanks for updating this PR. We checked the lines you've touched for PEP 8 issues, and found:

There are currently no PEP 8 issues detected in this Pull Request. Cheers! :beers:

Comment last updated at 2021-09-13 07:44:25 UTC